30 lines
640 B
JSON
30 lines
640 B
JSON
{
|
|
"estimator_name": "mlp",
|
|
"dataset_name": "random_xy_dataset",
|
|
"dataloader_name": "supervised_data_loader",
|
|
"estimator_kwargs": {
|
|
"input_dim": 4,
|
|
"output_dim": 2
|
|
},
|
|
"dataset_kwargs": {
|
|
"num_samples": 100000,
|
|
"preload": true,
|
|
"input_dim": 4,
|
|
"output_dim": 2
|
|
},
|
|
"dataset_split_fracs": {
|
|
"train": 0.4,
|
|
"val": 0.3,
|
|
"aux": [0.3]
|
|
},
|
|
"dataloader_kwargs": {
|
|
"batch_size": 16
|
|
},
|
|
"train_kwargs": {
|
|
"summarize_every": 20,
|
|
"max_epochs": 100,
|
|
"stop_after_epochs": 100
|
|
},
|
|
"load_only": false
|
|
}
|