search

StefOe / indrnn-pytorch

pytorch implementation of Independently Recurrent Neural Networks https://arxiv.org/abs/1803.04831
121 stars 33 forks source link

Version 0.3 #4

Closed sbrugman closed 6 years ago

sbrugman commented 6 years ago

There is one single change to make addition_test.py work with PyTorch 0.3.XX, which is:

# For PyTorch 0.3.XX
losses.append(loss.data.cpu())
# For PyTorch 0.4.XX
losses.append(loss.data.cpu().item())
StefOe commented 6 years ago

I want to abandon PyTorch 0.3. as the current stable branch is 0.4 and in 0.5/1.0 this wont work.