Update to reflect the interfaces in lightning 1.5 - Githubissues

LoicGrobol / zeldarose

Train transformer-based models.

https://zeldarose.readthedocs.io

Other

28 stars 3 forks source link

Update to reflect the interfaces in lightning 1.5 #14

Closed LoicGrobol closed 2 years ago

LoicGrobol commented 2 years ago

Changed

The command line options have been changed to reflect change in Lightning
- --accelerator is now used for devices, tested values are "cpu" and "gpu"
- --strategy now specifies how to train, tested values are None (missing), "ddp", "ddp_sharded" "ddp_spawn" and"ddp_sharded_spawn".
- No more option to select sharded training, use the strategy alias for that
- --n-gpus has been renamed to --num-devices.

Internal

Tests now run in Pytest using the console-scripts plugin for smoke tests.
Smoke tests now include ddp_spawn tests and tests on gpu devices if available.