Closed rginpan closed 1 year ago
Could you give more details about regularization loss in implementation?
Sorry for the late reply. I have uploaded the files for training in the dev branch (still messy...) and you may find the implementation of regularization loss in L564 and L568 in the trainer.py.
dev
trainer.py
Could you give more details about regularization loss in implementation?