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

fix black format

Browse Source
This commit is contained in:
Jack Morris
2020-06-25 14:33:13 -04:00
parent 25b83de8d4
commit f7d38f7668
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
textattack/commands/train_model/run_training.py
View File

@@ -263,8 +263,12 @@ def train_model(args):
loss.backward()
return loss
for epoch in tqdm.trange(int(args.num_train_epochs), desc="Epoch", position=0, leave=True):
prog_bar = tqdm.tqdm(train_dataloader, desc="Iteration", position=0, leave=False)
for epoch in tqdm.trange(
int(args.num_train_epochs), desc="Epoch", position=0, leave=True
):
prog_bar = tqdm.tqdm(
train_dataloader, desc="Iteration", position=0, leave=False
)
for step, batch in enumerate(prog_bar):
input_ids, labels = batch
labels = labels.to(device)