I need help with google colab

[R3dPhil]

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]

I got this error when running the training code btw

[bibidentuhanoi]

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

[1AakashK]

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