R2-V2 / bv_config.json
j-morano's picture
Add bv model (safetensors) with standalone inference pipeline
bb8c6c9
Raw
History Blame Contribute Delete
485 Bytes
{
"dataset": "Unified_Fundus",
"num_iterations": 5,
"criterion": "RRLoss",
"base_criterion": "MultiMxCrBCELoss",
"model": "RRWNet",
"num_folds": 0,
"learning_rate": 0.0001,
"num_epochs": null,
"base_channels": 64,
"in_channels": 6,
"out_channels": 3,
"gpu_id": 0,
"n_proc": 1,
"data_folder": "../_Processed_Data",
"version": "Journal_paper",
"seed": 77,
"save_period": 5,
"overwrite": false,
"use_cfp": true
}