search

THU-MIG / RepViT

RepViT: Revisiting Mobile CNN From ViT Perspective [CVPR 2024] and RepViT-SAM: Towards Real-Time Segmenting Anything
https://arxiv.org/abs/2307.09283
Apache License 2.0
730 stars 55 forks source link

What is the training parameters of RepViT while using Timm repository？ #15

Closed Anleeno-Xu closed 11 months ago

jameslahm commented 11 months ago

Thanks. You could refer to https://github.com/THU-MIG/RepViT/blob/8ef1a71e299fc02ea5eaf201039f310313a66b96/main.py#L30-L186 for all training parameters.

Anleeno-Xu commented 11 months ago

@jameslahm Thanks! How long does it take to train the model using 8 RTX3090 GPUs? Do I need to modify the training parameters using 4 GPUs?

jameslahm commented 11 months ago

How long does it take to train the model using 8 RTX3090 GPUs?

It costs ~34 hours to train RepViT-M1 using 8 RTX3090 GPUs.

Do I need to modify the training parameters using 4 GPUs?

No need.