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.

Sign up for GitHub

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

Jump to bottom

Few changes in cifar10-rotate/main.py #1

Open

Piyushi-0 wants to merge 2 commits into uclaml:main from Piyushi-0:main

Piyushi-0 commented Dec 12, 2023

Thanks for releasing the code. Following are a few small issues that I tried to fix in the file cifar10-rotate/main.py.

The trainloader & testloader contained (inputs, targets) only, but the code also tried to load 'clusters'. Reference in the case of testloader. Reference in the case of trainloader. This caused code to crash with an error. I have tried to fix this in the revised code. I also noted that the cluster variables are not used in the rest of the code.
'net' was not defined before loading the checkpoint. Reference. This was fixed in the revised code.

Thanks for your time.

Piyushi-0 added 2 commits

December 12, 2023 14:15


          removed unused clusters variable.

793a74e


          defining net before loading checkpoint

7193b2b

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet