Commit f2f233ad authored by Andrei-Claudiu Roibu's avatar Andrei-Claudiu Roibu 🖥
Browse files

changed y to FloatTensor, for backpropagation

parent 8dc1afd3
......@@ -170,7 +170,7 @@ class Solver():
for batch_index, sampled_batch in enumerate(dataloaders[phase]):
X = sampled_batch[0].type(torch.FloatTensor)
y = sampled_batch[1].type(torch.LongTensor)
y = sampled_batch[1].type(torch.FloatTensor)
# We add an extra dimension (~ number of channels) for the 3D convolutions.
X = torch.unsqueeze(X, dim= 1)
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment