GPU數目疑問，結果與源碼不一致

Cysu / dgd_person_reid

Domain Guided Dropout for Person Re-identification

http://arxiv.org/abs/1604.07528

231 stars 94 forks source link

Closed gakkitw closed 7 years ago

gakkitw commented 8 years ago

你好，我想請問用兩張GPU並行訓練的用意是可以增強訓練還是只是為了增加訓練速度？因為我用單張GPU(GTX Titan X)且沒動任何參數，跑出的結果cuhk03 top-1 高於paper的數值，我覺得不太合理，想請問你有什麼看法，謝謝！

Cysu commented 8 years ago

并行训练主要是为了加快速度，prototxt不改的话单卡多卡是等价的。结果高于paper这个是有可能的，一是因为近期修复了一个iter_size的bug；二是因为不同的test split结果会略有差异，需要多做几次取平均。

gakkitw commented 7 years ago

感謝你的回答，我也多做了幾次取平均，取得了一些不錯的成果。但我還想再請教你一個問題，如果我想加入其他的datasets是否可行?如果可行我該從哪邊下手?期待你的答覆，麻煩你了。

Cysu commented 7 years ago

加入其他datasets的話可以：

利用目前訓練好的model作爲起始，直接在新的datasets上進行finetune。
把新的datasets加入到joint training的過程中。以market1501爲例，首先需要寫一個data/format_market1501.py將原始數據轉換成統一的格式；然後修改對應的prototxt和bash scripts，比如總人數變了，最後fc層的num_output也要跟着變。