YAML
model:
type: "resnet"
num_layers: 18
num_classes: 10
optimizer:
type: "adam"
learning_rate: 0.001
weight_decay: 0.0001
training:
batch_size: 32
num_epochs: 100
early_stopping:
patience: 5
min_delta: 0.001
data:
train_data_path: "/path/to/train_data"
val_data_path: "/path/to/val_data"Last updated