Register
Login
Resources
Docs Blog Datasets Glossary Case Studies Tutorials & Webinars
Product
Data Engine LLMs Platform Enterprise
Pricing Explore
Connect to our Discord channel

#204 Feature/sg 131 check env compliance

Merged
GitHub User merged 1 commits into Deci-AI:master from deci-ai:feature/SG-131_check_env_compliance
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
  1. # This recipe is for testing purpose only
  2. defaults:
  3. - training_hyperparams: cifar10_resnet_train_params
  4. - arch_params: resnet18_cifar_arch_params
  5. - checkpoint_params: default_checkpoint_params
  6. dataset_interface:
  7. classification_test_dataset:
  8. dataset_params:
  9. batch_size: 10
  10. data_loader_num_workers: 1
  11. load_checkpoint: False
  12. checkpoint_params:
  13. load_checkpoint: ${load_checkpoint}
  14. experiment_name: test
  15. model_checkpoints_location: local
  16. sg_model:
  17. _target_: super_gradients.SgModel
  18. experiment_name: ${experiment_name}
  19. model_checkpoints_location: ${model_checkpoints_location}
  20. multi_gpu: Off
  21. architecture: resnet18
Discard
Tip!

Press p or to see the previous file or, n or to see the next file