updated ssd weights
parent
c53f9df2e4
commit
c4bf0ba1be
|
@ -3,8 +3,8 @@ import { getModelUris } from '../common/getModelUris';
|
||||||
import { fetchJson } from './fetchJson';
|
import { fetchJson } from './fetchJson';
|
||||||
export async function loadWeightMap(uri, defaultModelName) {
|
export async function loadWeightMap(uri, defaultModelName) {
|
||||||
const { manifestUri, modelBaseUri } = getModelUris(uri, defaultModelName);
|
const { manifestUri, modelBaseUri } = getModelUris(uri, defaultModelName);
|
||||||
const manifest = await fetchJson(manifestUri);
|
let manifest = await fetchJson(manifestUri);
|
||||||
console.log(typeof manifest, manifest);
|
// if (manifest['weightsManifest']) manifest = manifest['weightsManifest'];
|
||||||
return tf.io.loadWeights(manifest, modelBaseUri);
|
return tf.io.loadWeights(manifest, modelBaseUri);
|
||||||
}
|
}
|
||||||
//# sourceMappingURL=loadWeightMap.js.map
|
//# sourceMappingURL=loadWeightMap.js.map
|
|
@ -1 +1 @@
|
||||||
{"version":3,"file":"loadWeightMap.js","sourceRoot":"","sources":["../../src/dom/loadWeightMap.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,uBAAuB,CAAC;AAE5C,OAAO,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AACtD,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAExC,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,GAAuB,EACvB,gBAAwB;IAExB,MAAM,EAAE,WAAW,EAAE,YAAY,EAAE,GAAG,YAAY,CAAC,GAAG,EAAE,gBAAgB,CAAC,CAAA;IACzE,MAAM,QAAQ,GAAG,MAAM,SAAS,CAA8B,WAAW,CAAC,CAAA;IAC1E,OAAO,CAAC,GAAG,CAAC,OAAO,QAAQ,EAAE,QAAQ,CAAC,CAAC;IACvC,OAAO,EAAE,CAAC,EAAE,CAAC,WAAW,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAA;AAClD,CAAC"}
|
{"version":3,"file":"loadWeightMap.js","sourceRoot":"","sources":["../../src/dom/loadWeightMap.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,uBAAuB,CAAC;AAE5C,OAAO,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AACtD,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAExC,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,GAAuB,EACvB,gBAAwB;IAExB,MAAM,EAAE,WAAW,EAAE,YAAY,EAAE,GAAG,YAAY,CAAC,GAAG,EAAE,gBAAgB,CAAC,CAAA;IACzE,IAAI,QAAQ,GAAG,MAAM,SAAS,CAA8B,WAAW,CAAC,CAAA;IACxE,2EAA2E;IAC3E,OAAO,EAAE,CAAC,EAAE,CAAC,WAAW,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAA;AAClD,CAAC"}
|
|
@ -21539,8 +21539,7 @@ Manifest JSON has weights with names: ${allManifestWeightNames.join(", ")}.`);
|
||||||
// build/dom/loadWeightMap.js
|
// build/dom/loadWeightMap.js
|
||||||
async function loadWeightMap(uri, defaultModelName) {
|
async function loadWeightMap(uri, defaultModelName) {
|
||||||
const {manifestUri, modelBaseUri} = getModelUris(uri, defaultModelName);
|
const {manifestUri, modelBaseUri} = getModelUris(uri, defaultModelName);
|
||||||
const manifest = await fetchJson(manifestUri);
|
let manifest = await fetchJson(manifestUri);
|
||||||
console.log(typeof manifest, manifest);
|
|
||||||
return io_exports.loadWeights(manifest, modelBaseUri);
|
return io_exports.loadWeights(manifest, modelBaseUri);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
|
@ -8,6 +8,7 @@ export async function loadWeightMap(
|
||||||
defaultModelName: string,
|
defaultModelName: string,
|
||||||
): Promise<tf.NamedTensorMap> {
|
): Promise<tf.NamedTensorMap> {
|
||||||
const { manifestUri, modelBaseUri } = getModelUris(uri, defaultModelName)
|
const { manifestUri, modelBaseUri } = getModelUris(uri, defaultModelName)
|
||||||
const manifest = await fetchJson<tf.io.WeightsManifestConfig>(manifestUri)
|
let manifest = await fetchJson<tf.io.WeightsManifestConfig>(manifestUri)
|
||||||
|
// if (manifest['weightsManifest']) manifest = manifest['weightsManifest'];
|
||||||
return tf.io.loadWeights(manifest, modelBaseUri)
|
return tf.io.loadWeights(manifest, modelBaseUri)
|
||||||
}
|
}
|
|
@ -0,0 +1,3 @@
|
||||||
|
version https://git-lfs.github.com/spec/v1
|
||||||
|
oid sha256:6ff16f095b0371f24accb48aab82e595697baa821cf851167cbf9203d1d7ec9a
|
||||||
|
size 4194304
|
|
@ -1,3 +0,0 @@
|
||||||
version https://git-lfs.github.com/spec/v1
|
|
||||||
oid sha256:dcaf180d06ee1f621a3bd4bfec3783ddb155f0a47d782a5387b66afd126fcac9
|
|
||||||
size 4194304
|
|
|
@ -0,0 +1,3 @@
|
||||||
|
version https://git-lfs.github.com/spec/v1
|
||||||
|
oid sha256:2539847e627c8c4a561e999170735e6c527deecc6fdba2959fc5312fb1ea1de3
|
||||||
|
size 1422653
|
|
@ -1,3 +0,0 @@
|
||||||
version https://git-lfs.github.com/spec/v1
|
|
||||||
oid sha256:715a709e20a72e3d5c396b0caffab6f1218b5dd668081bf3c60951343e688075
|
|
||||||
size 4194304
|
|
|
@ -1,3 +0,0 @@
|
||||||
version https://git-lfs.github.com/spec/v1
|
|
||||||
oid sha256:b9eacbf6ce1c4600f5d4eb1ae36ad19a01d223f5b15b2106edea82fdd3cd4c3f
|
|
||||||
size 4194304
|
|
|
@ -1,3 +0,0 @@
|
||||||
version https://git-lfs.github.com/spec/v1
|
|
||||||
oid sha256:3ce73b2a6edde0d0fe4e027ab1f5e6aee68fa1908be58be79e9d06dda7fc3ef8
|
|
||||||
size 4194304
|
|
|
@ -1,3 +0,0 @@
|
||||||
version https://git-lfs.github.com/spec/v1
|
|
||||||
oid sha256:9648841c0475c6dc8ccbae63d412e34ee4a4f7d8474f65b1876e4adc9c8b5e06
|
|
||||||
size 4194304
|
|
|
@ -1,3 +0,0 @@
|
||||||
version https://git-lfs.github.com/spec/v1
|
|
||||||
oid sha256:0c6db8cacde568cd5e77f7abfb645bb50d5639bbced90dbc33e62d0b198c2181
|
|
||||||
size 1193002
|
|
|
@ -1,3 +1,3 @@
|
||||||
version https://git-lfs.github.com/spec/v1
|
version https://git-lfs.github.com/spec/v1
|
||||||
oid sha256:d5587814397f0a6f5064aa961bdae3e62093076f13342af405595996eafe8c1c
|
oid sha256:9b84918d1d8b2e988dc5d72c0d77e7cc0a3d433f2452516fcd88dca8051b552f
|
||||||
size 300234
|
size 26548
|
||||||
|
|
Loading…
Reference in New Issue