Closed NanCheng2001 closed 8 months ago
I did not remember the exact numbers. I think it should be <10GB for given batch size. The network is very light-weighted. Because the feature backbone is frozen and the rests are fine tuned but they do not have many parameters.
Thank you for your patience.
Thank you for your outstanding work!
I would like to ask, what are the equipment requirements for your model training, and at least how much GPUmemory is required? Because your article does not specify which parameters are frozen and which parameters need to be retuned.