Support pytorch's wrapped models in trainer.py

sh0416 / clrcmd

Official Repository for CLRCMD (Appear in ACL2022)

Other

38 stars 4 forks source link

Support pytorch's wrapped models in trainer.py #2

Closed atturaioe closed 1 year ago

atturaioe commented 1 year ago

Support wrapped models in trainer.py::STSTrainer::prediction_step. Using model.model is assumed that model is not wrapped in Pytorch's DataParallel layers.