init

{{cookiecutter.project_slug}}/{{cookiecutter.module_name}}/config/app.toml

@@ -0,0 +1,3 @@
+[app]
+host = "127.0.0.1"
+port = 8001

{{cookiecutter.project_slug}}/{{cookiecutter.module_name}}/config/data.toml

@@ -0,0 +1,4 @@
+[data]
+train_path = "/path/to/data/mnist_train.csv"
+in_channels = 1
+num_classes = 10

{{cookiecutter.project_slug}}/{{cookiecutter.module_name}}/config/model.toml

@@ -0,0 +1,3 @@
+[model]
+hidden_size = 8
+name = 'vgg11'

{{cookiecutter.project_slug}}/{{cookiecutter.module_name}}/config/paths.toml

@@ -0,0 +1,4 @@
+[paths]
+repo = "/path/to/root"
+app = "/path/to/root/ml_pipeline"
+data = "/path/to/root/data"

{{cookiecutter.project_slug}}/{{cookiecutter.module_name}}/config/training.toml

@@ -0,0 +1,8 @@
+[training]
+batch_size = 16
+epochs = 10
+learning_rate = 0.01
+device = 'cpu'
+# examples = 50
+examples = -1
+