-
Notifications
You must be signed in to change notification settings - Fork 6
Expand file tree
/
Copy pathlearning_params.toml
More file actions
47 lines (40 loc) · 856 Bytes
/
learning_params.toml
File metadata and controls
47 lines (40 loc) · 856 Bytes
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
DataDir = "${CoverageControl_ws}/lpac/" # Absolute location
GPUs = [0, 1]
NumWorkers = 4
# Directory to save the model
# If a model is already present, it will be loaded
# Similarly, for the optimizer
[LPACModel]
Dir = "${CoverageControl_ws}/lpac/models/"
# PreTrainedModel = "model.pt"
[CNNModel]
Dir = "${CoverageControl_ws}/lpac/models/" # Absolute location
Model = "model.pt"
Optimizer = "optimizer.pt"
[ModelConfig]
UseCommMaps = true
TargetType = "actions" # or goals
[GNNBackBone]
InputDim = 7
NumHops = 3
NumLayers = 5
LatentSize = 256
OutputDim = 2
[LPACTraining]
LearningRate = 0.0001
WeightDecay = 0.001
BatchSize = 10
NumEpochs = 15
[CNNBackBone]
InputDim = 4
NumLayers = 3
LatentSize = 32
KernelSize = 3
ImageSize = 32
OutputDim = 7
[CNNTraining]
LearningRate = 0.0001
WeightDecay = 0.001
BatchSize = 10
NumEpochs = 15
Momentum = 0.1