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
When I am trying to train the model with resnet101, I am getting the following error. Please help me figure this out.
0%| | 0/14848 [00:00<?, ?it/s]
Epoch 0: 0%| | 0/14848 [00:00<?, ?it/s]total images = 17469
create resnet101 model
Started training model from epoch 0
Traceback (most recent call last):
File "train_unet.py", line 251, in
train(train_loader, model, criterion, optimizer, validate, args)
File "train_unet.py", line 118, in train
masks_pred = model(input_var)
File "/opt/apps/Anaconda3/2019.03/envs/powerai16_ibm/lib/python3.6/site-packages/torch/nn/modules/module.py", line 489, in call
result = self.forward(*input, **kwargs)
File "/home/nys09/unet/unet_transfer.py", line 233, in forward
dec5 = self.dec5(torch.cat([center, conv5], 1))
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 1. Got 7 and 6 in dimension 2 at /opt/anaconda/conda-bld/pytorch_1551416914958/work/aten/src/THC/generic/THCTensorMath.cu:83
The text was updated successfully, but these errors were encountered:
When I am trying to train the model with resnet101, I am getting the following error. Please help me figure this out.
0%| | 0/14848 [00:00<?, ?it/s]
Epoch 0: 0%| | 0/14848 [00:00<?, ?it/s]total images = 17469
create resnet101 model
Started training model from epoch 0
Traceback (most recent call last):
File "train_unet.py", line 251, in
train(train_loader, model, criterion, optimizer, validate, args)
File "train_unet.py", line 118, in train
masks_pred = model(input_var)
File "/opt/apps/Anaconda3/2019.03/envs/powerai16_ibm/lib/python3.6/site-packages/torch/nn/modules/module.py", line 489, in call
result = self.forward(*input, **kwargs)
File "/home/nys09/unet/unet_transfer.py", line 233, in forward
dec5 = self.dec5(torch.cat([center, conv5], 1))
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 1. Got 7 and 6 in dimension 2 at /opt/anaconda/conda-bld/pytorch_1551416914958/work/aten/src/THC/generic/THCTensorMath.cu:83
The text was updated successfully, but these errors were encountered: