search

Cadene / vqa.pytorch

Visual Question Answering in Pytorch
717 stars 178 forks source link

Make training commands work #23

Closed ssnl closed 7 years ago

ssnl commented 7 years ago

Fixes several issues so that the training commands in README work:

  1. fix os.execute -> os.system
  2. fix hacks for data_parallel. They weren't working in data parallel because the old lambda function refers to tensors not transferred to proper GPU yet.
  3. fix train/val/test directory checking in COCO image loading. Current code only checks the containing dir of these split sets so it errors in cases when downloading val after train.
  4. fix training config yaml missing size for coco, and thus can't find the correct directory to load data.
Cadene commented 7 years ago

Thanks so much!