MCPcopy
hub / github.com/tensorflow/tfjs / compile

Method compile

tfjs-layers/src/models.ts:813–826  ·  view source on GitHub ↗

* See `LayersModel.compile`. * * @param args

(args: ModelCompileArgs)

Source from the content-addressed store, hash-verified

811 * @param args
812 */
813 override compile(args: ModelCompileArgs): void {
814 this.build();
815 this.model.compile(args);
816 this.optimizer_ = this.model.optimizer;
817 // tslint:disable-next-line:no-any
818 this.isOptimizerOwned = (this.model as any).isOptimizerOwned;
819 this.loss = this.model.loss;
820 this.metrics = this.model.metrics;
821 // TODO(cais): Add this.lossWeights, this.sampleWeightMode,
822 // this.weightedMetrics, this.targets.
823 this.metricsTensors = this.model.metricsTensors;
824 this.metricsNames = this.model.metricsNames;
825 // TODO(cais): Add sampleWeights.
826 }
827
828 override get optimizer(): Optimizer {
829 return this.model == null ? undefined : this.model.optimizer;

Calls 1

buildMethod · 0.95