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 19:20:54 -04:00
parent ba68897975
commit fb6a04e10b
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
textattack/trainer.py
View File

@@ -23,10 +23,9 @@ from .attacker import Attacker
from .model_args import HUGGINGFACE_MODELS from .model_args import HUGGINGFACE_MODELS
from .models.helpers import LSTMForClassification, WordCNNForClassification from .models.helpers import LSTMForClassification, WordCNNForClassification
from .models.wrappers import ModelWrapper from .models.wrappers import ModelWrapper
from .shared.utils.install import logger
from .training_args import CommandLineTrainingArgs, TrainingArgs from .training_args import CommandLineTrainingArgs, TrainingArgs
logger = logger logger = textattack.shared.utils.logger
class Trainer: class Trainer: