Update original.py
Browse files- original.py +1 -2
original.py
CHANGED
|
@@ -61,8 +61,7 @@ if config.dml == True:
|
|
| 61 |
|
| 62 |
fairseq.modules.grad_multiply.GradMultiply.forward = forward_dml
|
| 63 |
i18n = I18nAuto()
|
| 64 |
-
|
| 65 |
-
# 判断是否有能用来训练和加速推理的N卡
|
| 66 |
ngpu = torch.cuda.device_count()
|
| 67 |
gpu_infos = []
|
| 68 |
mem = []
|
|
|
|
| 61 |
|
| 62 |
fairseq.modules.grad_multiply.GradMultiply.forward = forward_dml
|
| 63 |
i18n = I18nAuto()
|
| 64 |
+
|
|
|
|
| 65 |
ngpu = torch.cuda.device_count()
|
| 66 |
gpu_infos = []
|
| 67 |
mem = []
|