Peter Sherman

Hey @yellowcap, I'm getting: `--------------------------------------------------------------------------- ValueError Traceback (most recent call last) Cell In[19], [line 147](vscode-notebook-cell:?execution_count=19&line=147) [137](vscode-notebook-cell:?execution_count=19&line=137) pixels = transform(pixels) [139](vscode-notebook-cell:?execution_count=19&line=139) datacube = ( [140](vscode-notebook-cell:?execution_count=19&line=140) torch.tensor(pixels, dtype=torch.float32, device="cuda"), [141](vscode-notebook-cell:?execution_count=19&line=141) torch.tensor(np.hstack((week_norm, hour_norm)),...

> @4242psherman4242 which pytorch version are you using to run the notebook? Hey @srmsoumya this is using version 2.1.2 of pytorch