Add initialized property for parallel wrappers

DrA1ex · Sep 20, 2023 · 57eea79 · 57eea79
1 parent 1f1e9e8
commit 57eea79
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/src/app/neural-network/engine/wrapper/parallel-gan.ts b/src/app/neural-network/engine/wrapper/parallel-gan.ts
@@ -18,6 +18,7 @@ export class ParallelGanWrapper {
     public readonly ganChainWrapper: ParallelModelWrapper<IModel>
 
     public get parallelism() {return this.generatorWrapper.parallelism;}
+    public get initialized() {return this.generatorWrapper.initialized;}
 
     constructor(public readonly gan: GenerativeAdversarialModel, parallelism?: number) {
         this.generatorWrapper = new ParallelModelWrapper(gan.generator, parallelism);

diff --git a/src/app/neural-network/engine/wrapper/parallel.ts b/src/app/neural-network/engine/wrapper/parallel.ts
@@ -111,6 +111,7 @@ export class ParallelModelWrapper<T extends IModel> {
     private _weights!: LayerWeights[];
 
     public readonly parallelism: number;
+    public get initialized() {return this._initialized;}
 
     constructor(public readonly model: T, parallelism = 4) {
         this.parallelism = Math.max(1, parallelism);