Closed HansBambel closed 1 year ago
This is a big refactor of the (kind of outdated) code.
The dependencies for almost all packages were updated. Most importantly, pytorch is now 2.0.0 and lightning uses 2.0.
I have not fully trained the models to check whether everything works correctly, but it is possible to execute the train_precip_lightning.py script.
train_precip_lightning.py
~test_precip_lightning.py was not tested yet.~
test_precip_lightning.py
Next things to do:
train_SmaAtUNet.py
This is a big refactor of the (kind of outdated) code.
The dependencies for almost all packages were updated. Most importantly, pytorch is now 2.0.0 and lightning uses 2.0.
I have not fully trained the models to check whether everything works correctly, but it is possible to execute the
train_precip_lightning.py
script.~
test_precip_lightning.py
was not tested yet.~Next things to do:
test_precip_lightning.py
train_SmaAtUNet.py