FazBrowse GitHub Viewer | Trending |
URL:
| Home
Tools: [Download Repo ZIP]   [Original HTTPS Page]

Refactor Outdated PyTorch Operations for AlexNet Training on CIFAR by KaledDahleh · Pull Request #57 · bearpaw/pytorch-classification · GitHub

Refactor Outdated PyTorch Operations for AlexNet Training on CIFAR - #57

Open
KaledDahleh wants to merge 5 commits into
bearpaw:masterfrom
KaledDahleh:master
Open

KaledDahleh wants to merge 5 commits into
bearpaw:masterfrom
KaledDahleh:master

Conversation

Copy link
Copy Markdown

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. Removed async=True argument from cuda() calls: This argument is deprecated in PyTorch 1.0.0 and has been removed to maintain compatibility.

    • Commit: 5c70d06
  2. Refactored code to use torch.no_grad() instead of volatile=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: Replaced view with reshape 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

This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters. Learn more about bidirectional Unicode characters
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant


Back | FazBrowse Home | New Git URL