ptrblck
/ pytorch_update_adam_internal_states
Created
November 26, 2018 22:13
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| """ | |
| Script to update parameters and the optimizer on the fly. | |
| It is not recommended to use this approach. | |
| This script just shows the disadvantages using this approach. | |
| @author: ptrblck | |
| """ | |
| import torch | |
| import torch.nn as nn |