Support optimizing/saving trainable parameters in the loss module

amjltc295 / pytorch-golden-template

PyTorch Golden Template (under development)

13 stars 4 forks source link

Support optimizing/saving trainable parameters in the loss module #76

Open ntubiolin opened 4 years ago

ntubiolin commented 4 years ago

Recently I am trying to use one optimizable loss function from "A General and Adaptive Robust Loss Function" Jonathan T. Barron, CVPR 2019 However, the current template seems not support the optimization of loss module. Will this template support this feature?