File tree Expand file tree Collapse file tree 3 files changed +9
-9
lines changed
template-vision-classification
template-vision-segmentation Expand file tree Collapse file tree 3 files changed +9
-9
lines changed Original file line number Diff line number Diff line change 1
1
# ::= from_template_common ::#
2
- lr : 0.0001
3
- model : resnet18
2
+ lr : 0.0001 # learning rate
3
+ model : resnet18 # model name
Original file line number Diff line number Diff line change 1
1
# ::= from_template_common ::#
2
- lr : 0.0001
3
- z_dim : 100
4
- d_filters : 64
5
- g_filters : 64
2
+ lr : 0.0001 # learning rate
3
+ z_dim : 100 # input dimension
4
+ d_filters : 64 # number of filters in the first conv layer of discriminator
5
+ g_filters : 64 # number of filters in the first conv layer of generator
Original file line number Diff line number Diff line change 1
1
# ::= from_template_common ::#
2
- lr : 0.007
3
- accumulation_steps : 4
4
- num_classes : 21
2
+ lr : 0.007 # learning rate
3
+ accumulation_steps : 4 # parameter for gradient accumulation
4
+ num_classes : 21 # number of classes for segmentation
You can’t perform that action at this time.
0 commit comments