loadWeightMap.js 1014 B

1234567891011121314151617181920212223
  1. "use strict";
  2. Object.defineProperty(exports, "__esModule", { value: true });
  3. var tslib_1 = require("tslib");
  4. var tf = require("@tensorflow/tfjs-core");
  5. var getModelUris_1 = require("../common/getModelUris");
  6. var fetchJson_1 = require("./fetchJson");
  7. function loadWeightMap(uri, defaultModelName) {
  8. return tslib_1.__awaiter(this, void 0, void 0, function () {
  9. var _a, manifestUri, modelBaseUri, manifest;
  10. return tslib_1.__generator(this, function (_b) {
  11. switch (_b.label) {
  12. case 0:
  13. _a = getModelUris_1.getModelUris(uri, defaultModelName), manifestUri = _a.manifestUri, modelBaseUri = _a.modelBaseUri;
  14. return [4 /*yield*/, fetchJson_1.fetchJson(manifestUri)];
  15. case 1:
  16. manifest = _b.sent();
  17. return [2 /*return*/, tf.io.loadWeights(manifest, modelBaseUri)];
  18. }
  19. });
  20. });
  21. }
  22. exports.loadWeightMap = loadWeightMap;
  23. //# sourceMappingURL=loadWeightMap.js.map