×

Welcome to TagMyCode

Please login or create account to add a snippet.
 

hydra_example_experiment.yaml

0
0
 
0
Language: Text
Posted by: Henry Lao
Added: Apr 15, 2022 3:39 AM
Views: 46
Tags: experimentation hydra ml pytorch template workflow
  1. # @package _global_
  2.  
  3. # to execute this experiment run:
  4. # python train.py experiment=example
  5.  
  6. defaults:
  7.   - override /datamodule: mnist.yaml
  8.   - override /model: mnist.yaml
  9.   - override /callbacks: default.yaml
  10.   - override /logger: null
  11.   - override /trainer: default.yaml
  12.  
  13. # all parameters below will be merged with parameters from default configurations set above
  14. # this allows you to overwrite only specified parameters
  15.  
  16. # name of the run determines folder name in logs
  17. name: "simple_dense_net"
  18.  
  19. seed: 12345
  20.  
  21. trainer:
  22.   min_epochs: 10
  23.   max_epochs: 10
  24.   gradient_clip_val: 0.5
  25.  
  26. model:
  27.   lr: 0.002
  28.   net:
  29.     lin1_size: 128
  30.     lin2_size: 256
  31.     lin3_size: 64
  32.  
  33. datamodule:
  34.   batch_size: 64
  35.  
  36. logger:
  37.   wandb:
  38.     tags: ["mnist", "${name}"]
  39.  

Write a comment

To post comments, please log in or register.