runtime error

Exit code: 1. Reason: ain-00001-of-00002.parquet: 100%|█████████▉| 361M/361M [00:01<00:00, 249MB/s] test-00000-of-00001.parquet: 0%| | 0.00/171M [00:00<?, ?B/s] test-00000-of-00001.parquet: 100%|█████████▉| 171M/171M [00:00<00:00, 255MB/s] Generating train split: 0%| | 0/43503 [00:00<?, ? examples/s] Generating train split: 100%|██████████| 43503/43503 [00:00<00:00, 46718.09 examples/s] Generating test split: 0%| | 0/10878 [00:00<?, ? examples/s] Generating test split: 100%|██████████| 10878/10878 [00:00<00:00, 59210.25 examples/s] /usr/local/lib/python3.10/site-packages/torchvision/models/_utils.py:208: UserWarning: The parameter 'pretrained' is deprecated since 0.13 and may be removed in the future, please use 'weights' instead. warnings.warn( /usr/local/lib/python3.10/site-packages/torchvision/models/_utils.py:223: UserWarning: Arguments other than a weight enum or `None` for 'weights' are deprecated since 0.13 and may be removed in the future. The current behavior is equivalent to passing `weights=None`. warnings.warn(msg) Traceback (most recent call last): File "/home/user/app/app.py", line 18, in <module> model.fc = nn.Linear(model.fc.in_features, NUM_CLASSES) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/linear.py", line 106, in __init__ torch.empty((out_features, in_features), **factory_kwargs) TypeError: empty() received an invalid combination of arguments - got (tuple, dtype=NoneType, device=NoneType), but expected one of: * (tuple of ints size, *, tuple of names names, torch.memory_format memory_format = None, torch.dtype dtype = None, torch.layout layout = None, torch.device device = None, bool pin_memory = False, bool requires_grad = False) * (tuple of ints size, *, torch.memory_format memory_format = None, Tensor out = None, torch.dtype dtype = None, torch.layout layout = None, torch.device device = None, bool pin_memory = False, bool requires_grad = False)

Container logs: