mirror of
https://github.com/velocitatem/cvfs.git
synced 2026-05-31 16:53:38 +00:00
Initial commit
This commit is contained in:
10
ml/configs/train/default.yaml
Normal file
10
ml/configs/train/default.yaml
Normal file
@@ -0,0 +1,10 @@
|
||||
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
|
||||
Reference in New Issue
Block a user