search

NJU-LHRS / official-CMID

The official implementation of paper "Unified Self-Supervised Learning Framework for Remote Sensing Images".
80 stars 5 forks source link

How long it will take？ #1

Closed guoya1003 closed 1 year ago

guoya1003 commented 1 year ago

Good job！！！ I just want to know how long it will take to train CMID on ResNet50 or Swin base using Million-AID or Potsdom datasets (how many 3090 or V100? A few days or a few hours?) Think you.

pUmpKin-Co commented 1 year ago

Hi~Thank you for following our work. Pre-training 200 epochs in MillionAID approximately takes 2 weeks (4 V100-32G with batch-size 64 x 4=256). Pre-training 400 epochs on Potsdam with ResNet50 approximately takes 10 hours (GTX3080Ti with batch-size 64). Pre-training 400 epochs on Potsdam with Swin-B approximately takes 36 hours (GTX3080Ti with batch-size 32).

pUmpKin-Co commented 1 year ago

Closed due to long periods of inactivity, can be reopened at any time if there is any problem.