master_thesis_models / configs /model /focusConv_150.yaml
Hannes Kuchelmeister
Add simple fully convolutional network
dce8df2
raw
history blame
228 Bytes
_target_: src.models.focus_conv_module.FocusConvLitModule
image_size: 150
pool_size: 2
conv1_size: 5
conv1_channels: 6
conv2_size: 5
conv2_channels: 16
lin1_size: 100
lin2_size: 80
output_size: 1
lr: 0.001
weight_decay: 0.0005