Update README.md

Update conv2d_gradfix.py

Update metric_utils.py

Update loss.py

Update training_loop.py

Merge pull request #10 from sklin93/main

adapt to newer pytorch versions

Update README.md

Update README.md

update

Update dataset_tool.py

Update generate.ipynb

Update README.md

Update README.md

Update loss.py