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
341 stars 62 forks source link

QUESTION #14

Closed dong-0412 closed 2 years ago

dong-0412 commented 2 years ago

Can I run this project with a single GPU? or what should change for run on single GPU?

RetroCirce commented 2 years ago

I update the code for single GPU usage. You can directly run with single GPU.