-
Notifications
You must be signed in to change notification settings - Fork 8
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
How to set the beta? #6
Comments
Hi @kojingying, Your For example, you can test if Afterwards, if you would prefer to use |
self.pool = nn.AvgPool2d(ih // 32, 1)<——self.pool = AdaPool2d(kernel_size=ih // 32, stride=1, beta=(1, 1)) When I try to replace avgpool2d with adapool2d, the following error occurs: “AssertionError: Only CUDA implementation supported!” How can I use it correctly? |
@Culturenotes it seems that either your input tensor or the beta is in the CPU. The current implementation only runs in CUDA, so you'll need to have all tensors used in |
Hi, thanks for your nice work! However, I am confusing on how to set the beta
The text was updated successfully, but these errors were encountered: