cs230-code-examples icon indicating copy to clipboard operation
cs230-code-examples copied to clipboard

Published 20 hours ago • verified publisher icon cs230-stanford

Batch norm layer fails when batch_size is set to 1

Open johntiger1 opened this issue 5 years ago • 0 comments

When running the code, I found that this particular layer (and only this one) fails when the batch size is set to 1.

https://github.com/cs230-stanford/cs230-code-examples/blob/159df10a6187c7e1d6ec949c8e06d7f67f8f1cd2/pytorch/vision/model/net.py#L78

johntiger1 avatar May 04 '19 02:05 johntiger1