javiribera / locating-objects-without-bboxes

PyTorch code for "Locating objects without bounding boxes" - Loss function and trained models
Other
249 stars 52 forks source link

Training epoch to get #36

Open futurianh1k opened 4 years ago

futurianh1k commented 4 years ago

Hi, I have been training with mall dataset, This is the command I used, python -m object-locator.train --train-dir ./object-locator/dataset/mall_dataset/frames/ \ --batch-size 32 \ --lr 1e-3 --optim Adam \ --resume ./model_08.ckpt --visdom-server localhost --visdom-port 8097 \ --val-dir auto \ --epochs 1000000 \ --nThreads 4 \ --val-freq 10 -\ -max-mask-pts 100 \ --paint

Could you please show me your train command, to get your achievement of getting head position? especially validation frequency and epoch setting.