-
Notifications
You must be signed in to change notification settings - Fork 48
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Batch size = 1 leads to error #28
Comments
The error comes from the global average pooling in the context module. The output of the global branch has no spatial resolution any more (see your error). If you are using batchsize 1, there is only one element left for determining mean and std in the subsequent batch normalization, which is just not possible. Note that this is requirement is only for training. Inference with batchsize 1 works well as the inference mode for batch normalization is different. |
I tried to train with default parameters and sunrgbd dataset with batch_size set to 1. The following error arises:
Any hint on how to fix it?
The text was updated successfully, but these errors were encountered: