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:47:36 -04:00
parent 841f47580b
commit ba68897975
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
textattack/trainer.py
View File

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