Open xshwen opened 7 years ago
Here is an example of using the optimizer interface: https://github.com/caffe2/caffe2/blob/master/caffe2/python/examples/resnet50_trainer.py#L289
You can just replace the sgd by adam:
optimizer.build_adam(model, learning_rate)
Have a look at optimizer.py for more details.
Where can I find a simple example of the use of adam?