search

cemanil / LNets

Lipschitz Neural Networks described in "Sorting Out Lipschitz Function Approximation" (ICML 2019).
55 stars 19 forks source link

Pytorch version 0.4 / 0.4.1? #1

Closed expectopatronum closed 5 years ago

expectopatronum commented 5 years ago

Hi, your README states "All the experiments were performed using PyTorch version 0.4" which made me think you used 0.4.0 but I think it should be 0.4.1 since for example from torch.utils.data import Subset, DataLoader causes an error using torch==0.4.0. torch.utils.data.dataset.Subset was moved to torch.utils.data.Subset in v0.4.1. It's working now for me but it might save others time if you could clarify this. Thanks and best regards, Verena

cemanil commented 5 years ago

Thank you for pointing this out. Updated the README accordingly.