Spaces:
Build error
Build error
# @package _global_ | |
# to execute this experiment run: | |
# python train.py experiment=example | |
defaults: | |
- override /datamodule: mnist.yaml | |
- override /model: mnist.yaml | |
- override /callbacks: default.yaml | |
- override /logger: null | |
- override /trainer: default.yaml | |
# all parameters below will be merged with parameters from default configurations set above | |
# this allows you to overwrite only specified parameters | |
# name of the run determines folder name in logs | |
name: "simple_dense_net" | |
seed: 12345 | |
trainer: | |
min_epochs: 10 | |
max_epochs: 10 | |
gradient_clip_val: 0.5 | |
model: | |
lr: 0.002 | |
net: | |
lin1_size: 128 | |
lin2_size: 256 | |
lin3_size: 64 | |
datamodule: | |
batch_size: 64 | |
logger: | |
wandb: | |
tags: ["mnist", "${name}"] |