search

kitzeslab / opensoundscape

Open source, scalable software for the analysis of bioacoustic recordings
http://opensoundscape.org
MIT License
138 stars 16 forks source link

support mixed precision in CNN #828

Closed sammlapp closed 1 month ago

sammlapp commented 8 months ago

example here: https://github.com/pscedu/bridges2-examples/blob/main/Bridges2-dist-DL-examples/Pytorch/pytorch_ddp.py

sammlapp commented 4 months ago

lightning branch's support for training with lightning.trainer enables mixed precision. We could still implement for plain pytorch if we want to

sammlapp commented 3 months ago

implemented in the lightning branch with .use_amp property, haven't tested on GPU yet

sammlapp commented 2 months ago

resolved in https://github.com/kitzeslab/opensoundscape/pull/1036