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

PyTorch Versions of TC1 and P1B2 Benchmarks #96

Open
wants to merge 2 commits into
base: develop
Choose a base branch
from

Commits on Aug 23, 2021

  1. PyTorch Versions of TC1 and P1B2 Benchmarks

    * baseline_pytorch.py is the main function of the benchmark.
    * Updated pytorch_utils.py to add additional functionality.
    * torch_deps folder consists of dependencies that are required for the benchmark to function correctly, including the model, dataset, classification network, training, validation, and test functions, and weight initialization.
    knangia04 committed Aug 23, 2021
    Configuration menu
    Copy the full SHA
    0109290 View commit details
    Browse the repository at this point in the history

Commits on Sep 18, 2021

  1. Updated to resolve comments on pull request

    1. Updated to use the original default_model txt file.
    2. Updated common/pytorch_utils.py based on the the local pytorch_utils.py and deleted the local versions.
    knangia04 committed Sep 18, 2021
    Configuration menu
    Copy the full SHA
    2654b2e View commit details
    Browse the repository at this point in the history