search

RetroCirce / HTS-Audio-Transformer

The official code repo of "HTS-AT: A Hierarchical Token-Semantic Audio Transformer for Sound Classification and Detection"
https://arxiv.org/abs/2202.00874
MIT License
344 stars 62 forks source link

training and evaluation on a single GPU #9

Closed joewale closed 2 years ago

joewale commented 2 years ago

HI, RetroCirce, I want to use a single GPU for training and evaluation. How to mannually change the code of sed_model.py and main.py?

 Looking forward to your reply. Thanks.
RetroCirce commented 2 years ago

Hi,

Here is the way to change it to single GPU: https://github.com/RetroCirce/HTS-Audio-Transformer/issues/5#issuecomment-1168168824

joewale commented 2 years ago

Thank you for your reply.