mirror of
https://github.com/velocitatem/cvfs.git
synced 2026-05-31 08:43:37 +00:00
11 lines
196 B
YAML
11 lines
196 B
YAML
input_dim: 16
|
|
hidden_dim: 64
|
|
num_classes: 3
|
|
learning_rate: 0.001
|
|
batch_size: 64
|
|
epochs: 5
|
|
log_every_n_steps: 20
|
|
seed: 42
|
|
tensorboard_dir: ml/tensorboard
|
|
weights_output: ml/models/weights/model.pt
|