SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

update requirements

Browse Source
This commit is contained in:
VictorSanh
2019-12-05 12:07:04 -05:00
parent 552c44a9b1
commit 35ff345fc9
2 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
examples/distillation/distiller.py
View File

@@ -21,7 +21,6 @@ import psutil
import time import time
from tqdm import trange, tqdm from tqdm import trange, tqdm
import numpy as np import numpy as np
import psutil
import torch import torch
import torch.nn as nn import torch.nn as nn

2
examples/distillation/requirements.txt
View File

@@ -3,4 +3,4 @@ tensorboard>=1.14.0
tensorboardX==1.8 tensorboardX==1.8
psutil==5.6.3 psutil==5.6.3
scipy==1.3.1 scipy==1.3.1
transformers==2.0.0 transformers