04.first-neural-network.ipynb error

[caxelrud]

I am getting the following error for 04.first-neural-network.ipynb notebook:

--------------------------------------------------
RuntimeError     Traceback (most recent call last)
<ipython-input-28-8756d5fbee9e> in <module>
     10 
     11     if epoch % 100 == 0:
---> 12       train_acc = calculate_accuracy(y_train, y_pred)
     13 
     14       y_test_pred = net(X_test)

<ipython-input-24-51ba3ab94870> in calculate_accuracy(y_true, y_pred)
      1 def calculate_accuracy(y_true, y_pred):
      2   predicted = y_pred.ge(.5).view(-1)
----> 3   return (y_true == predicted).sum().float() / len(y_true)

RuntimeError: Expected object of scalar type Float but got scalar type Byte for argument #2 'other'

[curiousily]

Hey @caxelrud,

Just rerun the notebook on Google Colab and everything seems to execute/train fine. Are you using different model/dataset/environment?