generative-inpainting-pytorch icon indicating copy to clipboard operation
generative-inpainting-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon daa233

Sync Batch norm not used?

Open dragen1860 opened this issue 4 years ago • 1 comments

Dear author: Thanks for your re-implementation, it's helpful! When I tried to train with multi gpus, I noticed you have not convert the batch norm to synced batch norm. I wonder if you realized it or not？ Thank you.

dragen1860 avatar May 19 '20 09:05 dragen1860

@dragen1860 Thanks for your attention!

The synced batch norm is not used in this repo. The original tensorflow implementation of deep fill v1 uses only 1 gpu to train the model. I extended this repo to support multiple gpus but not synced batch norm. You are welcomed to contribute the code!

daa233 avatar May 19 '20 11:05 daa233