deepspline
deepspline copied to clipboard
Fixed dataset_Reader bug where it returned labels of wrong dimensions
I apologize for the messy diff. My vim config autoformats python code. This PR fixes issue #1
Changed all occurences of data[0] to item() for new pytorch version Changed learning rate and weight decay to 1e-6 to avoid NaN losses