Skip to content

Commit 03c205e

Browse files
authored
Fix serialized classNames for Optimizers (tensorflow#1861)
BUG INTERNAL
1 parent 341a8be commit 03c205e

File tree

4 files changed

+4
-4
lines changed

4 files changed

+4
-4
lines changed

src/optimizers/adadelta_optimizer.ts

+1-1
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ import {Optimizer, OptimizerVariable} from './optimizer';
2525
/** @doclink Optimizer */
2626
export class AdadeltaOptimizer extends Optimizer {
2727
/** @nocollapse */
28-
static className = 'AdadeltaOptimizer';
28+
static className = 'Adadelta'; // Name matters for Python compatibility.
2929
private accumulatedGrads: OptimizerVariable[] = [];
3030
private accumulatedUpdates: OptimizerVariable[] = [];
3131

src/optimizers/momentum_optimizer.ts

+1-1
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ import {SGDOptimizer} from './sgd_optimizer';
2828
/** @doclink Optimizer */
2929
export class MomentumOptimizer extends SGDOptimizer {
3030
/** @nocollapse */
31-
static className = 'MomentumOptimizer';
31+
static className = 'Momentum'; // Name matters for Python compatibility.
3232
private m: Scalar;
3333
private accumulations: OptimizerVariable[] = [];
3434

src/optimizers/rmsprop_optimizer.ts

+1-1
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ import {Optimizer, OptimizerVariable} from './optimizer';
2626
/** @doclink Optimizer */
2727
export class RMSPropOptimizer extends Optimizer {
2828
/** @nocollapse */
29-
static className = 'RMSProp';
29+
static className = 'RMSProp'; // Note: Name matters for Python compatibility.
3030
private centered: boolean;
3131

3232
private accumulatedMeanSquares: OptimizerVariable[] = [];

src/optimizers/sgd_optimizer.ts

+1-1
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ import {Optimizer} from './optimizer';
2727
/** @doclink Optimizer */
2828
export class SGDOptimizer extends Optimizer {
2929
/** @nocollapse */
30-
static className = 'SGD'; // Name matters for Python compatibility.
30+
static className = 'SGD'; // Note: Name matters for Python compatibility.
3131
protected c: Scalar;
3232

3333
constructor(protected learningRate: number) {

0 commit comments

Comments
 (0)