search

longcw / yolo2-pytorch

YOLOv2 in PyTorch
1.54k stars 421 forks source link

training problems #87

Open chenhongming opened 6 years ago

chenhongming commented 6 years ago
  1. When running train.py,how can I select gpu_id,for 0,1,2....?
  2. What does batch_size and train_batch_size mean? 2018-05-09 5 52 40

Thank you for your help!!!

bigbrother33 commented 5 years ago

batch_size is for demo?