You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
This is not an error. If you want to use low precision, i.e., fp16, please install the apex with cuda support (https://github.com/NVIDIA/apex) and update pytorch to 1.0
[Resize(size=(256, 128), interpolation=PIL.Image.BICUBIC), Pad(padding=10, fill=0, padding_mode=constant), RandomCrop(size=(256, 128), padding=None), RandomHorizontalFlip(p=0.5), ToTensor(), Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])]
Traceback (most recent call last):
File "train.py", line 166, in
for x in ['train', 'val']}
File "train.py", line 166, in
for x in ['train', 'val']}
TypeError: init() got an unexpected keyword argument 'prefetch_factor'
The text was updated successfully, but these errors were encountered:
Hi @Schanppi
prefetch_factor is introduced to the latest pytorch (I guess since pytorch 1.6/1.7 or something like that).
I am sure 1.9 has prefetch_factor
This is not an error. If you want to use low precision, i.e., fp16, please install the apex with cuda support (https://github.com/NVIDIA/apex) and update pytorch to 1.0
[Resize(size=(256, 128), interpolation=PIL.Image.BICUBIC), Pad(padding=10, fill=0, padding_mode=constant), RandomCrop(size=(256, 128), padding=None), RandomHorizontalFlip(p=0.5), ToTensor(), Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])]
Traceback (most recent call last):
File "train.py", line 166, in
for x in ['train', 'val']}
File "train.py", line 166, in
for x in ['train', 'val']}
TypeError: init() got an unexpected keyword argument 'prefetch_factor'
The text was updated successfully, but these errors were encountered: