user7867665
user7867665

Reputation: 882

Optimizer with train_on_batch?

Is there any point of using an optimizer like adam if I only use train_on_batch()?

I do the training in a loop manually (for adversarial training) but I want the learning rate to be automatically adjusted.

Upvotes: 3

Views: 599

Answers (1)

Alexandre Passos
Alexandre Passos

Reputation: 5206

Even if you only use train_on_batch the same optimizer instance will be reused in tensorflow, so the stats should be correctly tracked.

Upvotes: 2

Related Questions