Skip to content

Commit 794ff37

Browse files
authored
Annotation for template-vision-* (#326)
Annotation for template-vision-classification,dcgan and segmentation
1 parent 39ef15c commit 794ff37

File tree

3 files changed

+9
-9
lines changed

3 files changed

+9
-9
lines changed
Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,3 @@
11
#::= from_template_common ::#
2-
lr: 0.0001
3-
model: resnet18
2+
lr: 0.0001 # learning rate
3+
model: resnet18 # model name
Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
#::= 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
Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
11
#::= 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

0 commit comments

Comments
 (0)