From 0fd3884a4e890a1b2991a9da5751021e9e54493f Mon Sep 17 00:00:00 2001
From: Janne Spijkervet <janne.spijkervet@gmail.com>
Date: Thu, 12 Mar 2020 19:43:16 +0100
Subject: [PATCH] Added seed to config

---
 config/config.yaml | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/config/config.yaml b/config/config.yaml
index 2a06b39..e10987d 100644
--- a/config/config.yaml
+++ b/config/config.yaml
@@ -1,4 +1,5 @@
 # train options
+seed: 42 # sacred handles automatic seeding when passed in the config
 batch_size: 256
 workers: 16
 start_epoch: 0
@@ -6,11 +7,12 @@ 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