Added seed to config

Spijkervet · Mar 12, 2020 · 0fd3884 · 0fd3884
1 parent 5aeb144
commit 0fd3884
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/config/config.yaml b/config/config.yaml
@@ -1,16 +1,18 @@
 # train options
+seed: 42 # sacred handles automatic seeding when passed in the config
 batch_size: 256
 workers: 16
 start_epoch: 0
 epochs: 100
 dataset: "STL10" # STL10
 
 # model options
-resnet: "resnet18"
+resnet: "resnet50"
 normalize: True
 projection_dim: 64 # "[...] to project the representation to a 128-dimensional latent space"
 
 # loss options
+optimizer: "Adam" # or LARS (experimental)
 temperature: 0.5 # see appendix B.7.: Optimal temperature under different batch sizes
 
 # reload options