GPU used for training TESTR

mlpc-ucsd / TESTR

(CVPR 2022) Text Spotting Transformers

Apache License 2.0

179 stars 22 forks source link

GPU used for training TESTR #18

Closed k-styles closed 1 year ago

k-styles commented 1 year ago

Hi, thanks for this research paper, pretty amazing work! I am trying to reproduce your results on Nvidia Tesla V100 32GB, but I ran into CUDA out of memory error. Could you please let me know, what GPU was used for this research work?

Command used for training on CTW1500 dataset:

python tools/train_net.py --config-file configs/TESTR/CTW1500/Base-CTW1500.yaml --num-gpus 1

zx1239856 commented 1 year ago

Try reducing SOLVER.IMS_PER_BATCH