ml-tools/config.json at master · msc5/ml-tools · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
{
	"datasets": {
		"omniglot": {
			"structure": ["Split", "Language", "Character", "Image"],
			"class_level": 2,
			"channels": 1,
			"path": "datasets/omniglot"
		},
		"miniimagenet": {
			"structure": ["Split", "Class", "Image"],
			"class_level": 1,
			"channels": 3,
			"path": "datasets/miniimagenet"
		},
		"dummy": {
			"structure": ["Split", "Class", "Subclass", "Image"],
			"class_level": 2,
			"channels": 1,
			"path": "datasets/dummy"
		}
	},
	"models": {
		"debug": {
			"arch": "CustomNetwork",
			"loss_fn": "MSE",
			"meta_layers": 3
		},
		"mn_debug": {
			"arch": "MatchingNetwork",
			"loss_fn": "MSE"
		}
	},
	"experiments": {
		"fewshot": {
			"dataset": "omniglot",
			"todo": "train",
			"batch_size": 20,
			"k": 5,
			"n": 1,
			"m": 1,
			"epochs": 250,
			"learning_rate": 0.001,
			"schedule": [50, 100, 200]
		}
	},
	"todo": [["debug", "fewshot"]]
}