search

KovenYu / MAR

Pytorch code for our CVPR'19 (oral) work: Unsupervised person re-identification by soft multilabel learning
https://kovenyu.com/publication/2019-cvpr-mar/
315 stars 83 forks source link

question about batchsize #15

Closed xmengxin closed 5 years ago

xmengxin commented 5 years ago

In main.py line 13, I'm confused about args.batch_size//2, why devide by 2？ The batchsize is set to 368， actually is 184 when it is devided by 2. Whether we can set batchsize=184 immediately.

KovenYu commented 5 years ago

@9meng9 target images and source (auxiliary) images are both needed during training. so we simply use a 5/5 division for them. args.batchsize specifies the overall batch size to fit into gpu memory.