Closed mhusseinsh closed 5 years ago
I believe this was a problem in failure of the download of vgg16.t7 I managed to download it manually and put it in the model folder
Now I have this error
File "/mnt/DTAA_data/DTAA/code/z637177/UNIT-master/utils.py", line 244, in vgg_preprocess
batch = batch.sub(Variable(mean)) # subtract mean
RuntimeError: Expected object of type torch.cuda.FloatTensor but found type torch.FloatTensor for argument #3 'other'
Update:
I had to change mean = tensortype(batch.data.size()).cuda()
to mean = tensortype(batch.data.size()).cuda()
in def vgg_preprocess(batch)
in utils.py
I am not sure if this is the correct solution
@mhusseinsh Thanks. This seems to fix the problem.
Hello, I followed your instructions on getting the network running on my own dataset. However, I receive an error
Any idea what can be the problem here ?