Handwritten_Digit_Classifier / saved_models /tiny_vgg_less_compute_settings.yaml
Jechen00's picture
changed model used by app to have less compute
b1814f3
raw
history blame contribute delete
243 Bytes
mod_kwargs:
fc_hidden_dim: 128
hidden_channels: 5
in_channels: 1
num_blks: 2
num_classes: 10
num_convs: 2
train_kwargs:
batch_size: 100
learning_rate: 0.001
min_delta: 0.0001
num_epochs: 300
num_workers: 0
patience: 50