prototypical-network / config.json
lif31up's picture
add: FewShotEpisoder and get_prototypes
918349d
raw
history blame contribute delete
257 Bytes
{
"sate": "model.state_dict()",
"FRAMEWORK": [5, 5, 2],
"MODEL_CONFIG": [3, 26, 3],
"HYPER_PARAMETERS": {"lr": 0.0001, "weight_decay": 0.0001},
"TRANSFORM": "transform",
"GET_PROTOTYPES": "get_prototypes(support_set)",
"METRIC": "euclidean"
}