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

type of GPU #50

Open haloolahh opened 10 months ago

haloolahh commented 10 months ago

What type of GPU was used for training and how long it was trained

RetroCirce commented 9 months ago

8 V100 GPU for 1-2 days training can lead to the reported performance. If you use one GPU, perhaps also can achieve it by 5-7 days training.

TFEI-Nagato commented 5 months ago

I only have one rtx4070ti, 12g graphics memory. Is the graphics memory sufficient to support the project?