search

krisrs1128 / clouds_dist

Simulation of low-clouds, from weather measures.
4 stars 1 forks source link

self.data in LowClouds.__getitem__ are 2D numpy arrays instead of 3D torch tensors #162

Open jassiene-mila opened 4 years ago

jassiene-mila commented 4 years ago

In these two lines, self.data is a 2D NumPy array and the self.transform function is expecting 3D torch tensors. A quick fix was to replace these lines with 

"metos": torch.tensor(self.data["metos"][i]).unsqueeze(0),
 "real_imgs": torch.tensor(self.data["real_imgs"][i]).unsqueeze(0),