fix for pytorch_lightning >= 1.0.0

AntixK / PyTorch-VAE

A Collection of Variational Autoencoders (VAE) in PyTorch.

Apache License 2.0

6.62k stars 1.06k forks source link

fix for pytorch_lightning >= 1.0.0 #21

Open GPla opened 3 years ago

lucasfijen commented 3 years ago

Nice pullrequest, following on your changes, I found that also default_save_path should be changed into default_root_dir for the Trainer in run.py to work with lightning 1.2.0

lukesalamone commented 3 years ago

Shouldn't this also change requirements.txt?