关于训练时间 - Githubissues

zhengchen1999 / CAT

PyTorch code for our NeurIPS 2022 paper "Cross Aggregation Transformer for Image Restoration"

Apache License 2.0

121 stars 8 forks source link

Open ZZcun opened 9 months ago

761qgmpgz943 commented 7 months ago

同问作者花了多长时间训练一个模型

zhengchen1999 commented 7 months ago

不好意思，之前的问题没看明白。在4卡A100下的训练时间： x2：大约4天 x3、x4：大约2天（利用x2预训练模型初始化，训练量减半）