mirror of
https://github.com/wassname/Castor.git
synced 2026-09-11 11:51:23 +08:00
Victor Yang
51d8e29525
* update nce-sm * refactor code, update torchtext * use shared evaluation * refactor code, use shared data loader * refactor code * refactor code * refactor code according to Michael's great suggestions * update readme and requirement * update datasets and readme * update data loader * add space between + * update refactor code * add nce-mp * remove duplicate files * update readme, refactor code according to mp_cnn and delete duplicate code, follow PEP8 standard * refactor code, add/delete comments * import exit from sys
8 lines
346 B
Python
8 lines
346 B
Python
from nce.nce_pairwise_mp.trainers.qa_trainer import QATrainer
|
|
|
|
|
|
class WikiQATrainerNCE(QATrainer):
|
|
|
|
def __init__(self, model, train_loader, trainer_config, train_evaluator, test_evaluator, dev_evaluator=None):
|
|
super(WikiQATrainerNCE, self).__init__(model, train_loader, trainer_config, train_evaluator, test_evaluator, dev_evaluator)
|