Pytorch implementation of learned optimizer (research in progress)
- Python 3.3+.
- PyTorch 1.0+.
- Tensorpack(https://github.com/tensorpack/tensorpack)
pip install --upgrade git+https://github.com/tensorpack/tensorpack.git
# or add `--user` to install to user's local directories
Example(which can be changed):
python main.py --problem=imagenet --train_optim=obsrv_single --save_dir=./savepath