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
I replace torch.save(filename, model:clearState():get(1)) with torch.save(filename, model:clearState()) when training decoder,
and everything goes fine.
(--nGPU is set 1)
Hi,
I've trained an encoder and a decoder, and the training configuration as following:
Encoder:
Decoder:
I used following script to try to forward a single image in the CamVid dataset,
, and I got the following error message:
My question is
Is there any mistakes when training encoder or decorder?
Why "print(model)" only show the part of the network?
(I DO NOT modify any lua files)
It seems the image is loaded succesfully,
and I also create a 4-D tensor for the input.
What's going wrong?
Many thanks.
The text was updated successfully, but these errors were encountered: