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

Added support of multiple test loaders in train_from_config #1641

Merged
merged 6 commits into from
Nov 15, 2023

Conversation

BloodAxe
Copy link
Contributor

How it supposed to look in YAML file

defaults:
  - cifar10_resnet

test_dataloaders:
  cifar10: cifar10_val
  cifar100: cifar100_val

dataset_params:
  test_dataset_params:
    cifar10:
      root: ./data/cifar10
      train: False
      transforms:
        - Resize:
            size: 32
        - ToTensor
        - Normalize:
            mean:
              - 0.4914
              - 0.4822
              - 0.4465
            std:
              - 0.2023
              - 0.1994
              - 0.2010
      target_transform: null
      download: True

    cifar100:
      root: ./data/cifar100
      train: False
      transforms:
        - Resize:
            size: 32
        - ToTensor
        - Normalize:
            mean:
              - 0.4914
              - 0.4822
              - 0.4465
            std:
              - 0.2023
              - 0.1994
              - 0.2010
      target_transform: null
      download: True

hydra:
  searchpath:
    - pkg://super_gradients.recipes

Copy link
Contributor

@shaydeci shaydeci left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@BloodAxe BloodAxe merged commit 7ff90e4 into master Nov 15, 2023
7 checks passed
@BloodAxe BloodAxe deleted the feature/SG-1210-add-test-loaders branch November 15, 2023 11:24
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.

2 participants