leolyj / IRAST

This is the code for the ECCV2020 paper "Semi-Supervised Crowd Counting via Self-Training on Surrogate Tasks"
5 stars 3 forks source link

GPU memory is required #2

Open youth-tang opened 2 years ago

youth-tang commented 2 years ago

Hello! Thank you so much for providing such excellent code. How much GPU memory is required for this code training? Thank you in advance and look forward to your reply.

Liuyan9527zns commented 2 years ago

We run the training stage on a 1080, 10GB GPU memory is enough.