search

pengzhiliang / Conformer

Official code for Conformer: Local Features Coupling Global Representations for Visual Recognition
Apache License 2.0
531 stars 87 forks source link

CUDA out of memory #5

Closed Sunting78 closed 3 years ago

Sunting78 commented 3 years ago

V100 32G run base model

pengzhiliang commented 3 years ago

Emm, you can try to reduce the batch size and reduce learning rate accordingly