alihan/textattack-nlp-transformer
1
0
Fork 0
mirror of https://github.com/QData/TextAttack.git synced 2021-10-13 00:05:06 +03:00
Code Issues Projects Releases Wiki Activity

Update trainer.py

Browse Source
This commit is contained in:
Hanyu Liu
2021-10-05 18:39:00 -04:00
parent 9b089994a1
commit 841f47580b
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
textattack/trainer.py
View File

@@ -25,8 +25,9 @@ from .models.helpers import LSTMForClassification, WordCNNForClassification
from .models.wrappers import ModelWrapper
from .training_args import CommandLineTrainingArgs, TrainingArgs
import textattack.shared as util_shared
from .shared.utils.install import logger as the_logger
logger = util_shared.logger
logger = the_logger
class Trainer: