Movatterモバイル変換


[0]ホーム

URL:


Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

Sign up
Appearance settings

Refactor Outdated PyTorch Operations for AlexNet Training on CIFAR#57

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 ourterms of service andprivacy statement. We’ll occasionally send you account related emails.

Already on GitHub?Sign in to your account

Open
KaledDahleh wants to merge5 commits intobearpaw:master
base:master
Choose a base branch
Loading
fromKaledDahleh:master

Conversation

@KaledDahleh
Copy link

This pull request includes several important updates and fixes to ensure compatibility with the latest version of PyTorch and to resolve issues related to tensor operations while training AlexNet on the CIFAR dataset.

The key changes are as follows:

  1. Removedasync=True argument fromcuda() calls: This argument is deprecated in PyTorch 1.0.0 and has been removed to maintain compatibility.

    • Commit:5c70d06
  2. Refactored code to usetorch.no_grad() instead ofvolatile=True: Updated to the new context manager approach to handle operations without tracking gradients for better efficiency with the latest PyTorch version.

    • Commit:4d393ff
  3. Fixed tensor reshaping in the accuracy function: Replacedview withreshape to correct issues when reshaping tensors, improving reliability.

    • Commit:d378d87
  4. Addressed all 0-dim tensor indexing errors in cifar.py: Accessing 0-dim tensors caused index errors, ensuring smooth operation during training and metrics update.

    • Commits:c570bd6,2935578

Additional Information:

  • Files Changed:
    • cifar.py: Updated to remove deprecated or obsolete arguments and fix tensor operations
    • eval.py: Adjusted tensor reshaping logic for accuracy calculation

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

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

1 participant

@KaledDahleh

[8]ページ先頭

©2009-2025 Movatter.jp