federer/dist/src/coordinator/optimizer.js

Experiments in asynchronous federated learning and decentralized learning

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.getOptimizer = void 0;
const tslib_1 = require("tslib");
const tf = tslib_1.__importStar(require("@tensorflow/tfjs-node"));
/** Creates the optimizer described by {@link OptimizerOptions} */
function getOptimizer(options) {
    switch (options.name) {
        case "sgd": {
            return tf.train.sgd(options.learningRate ?? 0.1);
        }
        case "adam": {
            return tf.train.adam(options.learningRate, options.beta1, options.beta2, options.epsilon);
        }
    }
}
exports.getOptimizer = getOptimizer;
//# sourceMappingURL=optimizer.js.map