deviandice/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix for torch2.0.0 suddenly being imposed on us

Browse Source
This commit is contained in:
mrq 2023-03-15 19:24:35 +00:00
parent b253da6e35
commit bea6174a19
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
codes/utils/util.py
View File

@ -22,7 +22,13 @@ from shutil import get_terminal_size
import scp import scp
import paramiko import paramiko
from torch.utils.checkpoint import checkpoint from torch.utils.checkpoint import checkpoint
from torch._six import inf
try:
# 1.13.1
from torch._six import inf
except Exception as e:
# 2.0
from torch import inf
import yaml import yaml