-
Change the methods to get the models, using the fully register methods
-
Change the methods to get loss functions. For the custom loss classes, we directly register them in the registry. And for the
PyTorch
loss classes, we wrap them into a new class in order to realize the registry mechanism (Not yet) -
Re-design the
coinfig/defaults.yaml
, add more flexibility to this file and make some setting more reasonable -
Add thePyPi
install method
This repository has been archived by the owner on Feb 17, 2022. It is now read-only.