Skip to content
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

[BUG] Adaptive Freezing doesn't work correctly with CIFAR10 and ResNet18 #89

Closed
baochunli opened this issue Jul 16, 2021 · 0 comments
Closed
Assignees
Labels
bug Something isn't working

Comments

@baochunli
Copy link
Collaborator

Describe the bug

The adaptive freezing example (examples/adaptive_freezing) does not work correctly with the CIFAR10 dataset and the ResNet18 model.

To Reproduce

python examples/adaptive_freezing.py, but uses CIFAR10 and ResNet18 in the .yml config file (examples/adaptive_freezing_CIFAR10_resnet18.yml).

Execution terminates with error:

line 114, in update_sync_mask
self.moving_average_deltas[name][indices], deltas[indices])
IndexError: too many indices for tensor of dimension 0

Expected behavior

Run completed successfully without errors.

@baochunli baochunli added the bug Something isn't working label Jul 16, 2021
@baochunli baochunli self-assigned this Jul 16, 2021
baochunli added a commit that referenced this issue Jul 16, 2021
… RFCs; Addressed an issue when running the adaptive freezing example with the ResNet18 model. Closed #89.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working
Projects
None yet
Development

No branches or pull requests

1 participant