@epfml/discojs 2.2.1 → 3.0.0

package/dist/training/trainer/trainer_builder.js

@@ -1,47 +0,0 @@
-import { DistributedTrainer } from './distributed_trainer.js';
-import { LocalTrainer } from './local_trainer.js';
-/**
- * A class that helps build the Trainer and auxiliary classes.
- */
-export class TrainerBuilder {
-    memory;
-    task;
-    constructor(memory, task) {
-        this.memory = memory;
-        this.task = task;
-    }
-    /**
-     * Builds a trainer object.
-     *
-     * @param client client to share weights with (either distributed or federated)
-     * @param distributed whether to build a distributed or local trainer
-     * @returns
-     */
-    async build(client, distributed = false) {
-        const model = await this.getModel(client);
-        if (distributed) {
-            return new DistributedTrainer(this.task, this.memory, model, client);
-        }
-        else {
-            return new LocalTrainer(this.task, this.memory, model);
-        }
-    }
-    /**
-     * If a model exists in memory, load it, otherwise load model from server
-     * @returns
-     */
-    async getModel(client) {
-        const modelID = this.task.trainingInformation?.modelID;
-        if (modelID === undefined) {
-            throw new TypeError('model ID is undefined');
-        }
-        const info = {
-            type: 'working',
-            taskID: this.task.id,
-            name: modelID,
-            tensorBackend: 'gpt'
-        };
-        const model = await (await this.memory.contains(info) ? this.memory.getModel(info) : client.getLatestModel());
-        return model;
-    }
-}