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.

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

Fix cifar10 model #270

Merged
merged 5 commits into from Jan 18, 2021
Merged

Fix cifar10 model #270

merged 5 commits into from Jan 18, 2021

Commits on Jan 17, 2021

  1. cifar10: replace CUDAapi by CUDA

    Joost committed Jan 17, 2021
    Configuration menu
    Copy the full SHA
    5e613b8 View commit details
    Browse the repository at this point in the history

  2. cifar10: switch dataset source and fix OOM errors 

    Dataset is switched from Metalhead to MLDatasets, and data is now moved
to the GPU in batches during training, validation and testing to prevent
OOM errors on less high end GPU's.
    Joost committed Jan 17, 2021
    Configuration menu
    Copy the full SHA
    9abb942 View commit details
    Browse the repository at this point in the history

Commits on Jan 18, 2021

  1. actually use testdata

    @joostdup
    joostdup committed Jan 18, 2021
    Configuration menu
    Copy the full SHA
    e0c37e3 View commit details
    Browse the repository at this point in the history

  2. Check for gpu and use device

    @joostdup
    joostdup committed Jan 18, 2021
    Configuration menu
    Copy the full SHA
    61440be View commit details
    Browse the repository at this point in the history

  3. cifar10: small fix and more feedback

    @joostdup
    joostdup committed Jan 18, 2021
    Configuration menu
    Copy the full SHA
    e888cbe View commit details
    Browse the repository at this point in the history