pytorch_tempest
pytorch_tempest copied to clipboard
Erlemar
My repo for training neural nets using pytorch-lightning and hydra
It is always good to have more options to choose. So it would be a good idea to add more schedulers. The steps are the following: - in conf/scheduler add...
It is always good to have more options to choose. So it would be a good idea to add more optimizers. The steps are the following: * in conf/optimizer add...
Add documentation with sphinx Basic docs are done, now it would be great, if they were automatically generated over all the code.
Often it is a good idea to train model on stages - for example with changes of loss, augs or other things. Need to implement it and be able to...