152334H / DL-Art-School

TorToiSe fine-tuning with DLAS
GNU Affero General Public License v3.0
214 stars 96 forks source link

I need help with google colab #83

Open R3dPhil opened 1 year ago

R3dPhil commented 1 year ago

Errormessage: /content/DL-Art-School/codes Traceback (most recent call last): File "/content/DL-Art-School/codes/train.py", line 12, in from data.data_sampler import DistIterSampler File "/content/DL-Art-School/codes/data/init.py", line 6, in from utils.util import opt_get File "/content/DL-Art-School/codes/utils/util.py", line 25, in from torch._six import inf ModuleNotFoundError: No module named 'torch._six'

R3dPhil commented 1 year ago

I got this error when running the training code btw

bibidentuhanoi commented 11 months ago

I fixed this problem by removing the torch._six import inf in /content/DL-Art-School/codes/utils/util.py

1AakashK commented 6 months ago

I fixed this by updating from torch._six import inf to from torch import inf. In line 25 of /content/DL-Art-School/codes/utils/util.py