16 lines
325 B
JSON
16 lines
325 B
JSON
{
|
|
"epochs": 4,
|
|
"embedding_dims": 128,
|
|
"batch_size": 32,
|
|
"classes" : ["black_red", "green_orange", "yellow_grey"],
|
|
"lr": 0.001,
|
|
"triplet-loss-margin": 1.0,
|
|
"triplet-loss-p": 2,
|
|
"train_size": 0.9,
|
|
"augmentations": true,
|
|
"env": "",
|
|
"path_to_dataset": "./triplet_dataset",
|
|
"visualize": true,
|
|
"pca_components": 16
|
|
}
|