File size: 1,743 Bytes
df370c5 | 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 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 | {
"evaluation": {
"batch_size": 1024,
"num_workers": 8,
"device": "auto",
"forward_batch_size": 64,
"cuda_autocast": true,
"cudnn_enabled": true,
"cudnn_benchmark": true,
"cudnn_deterministic": false
},
"loss": {
"name": "multi_positive_info_nce_with_decoy",
"temperature": 0.07,
"threshold": 0.99,
"decoy_weight": 1.0
},
"models": {
"SpectraVision-1C-CNN": {
"encoding": "SpectraVision-1C",
"encoding_params": {
"max_mz": 5000,
"num_peaks": 200,
"bin_multiply_factor": 10
},
"model_params": {
"img_size": [224, 224]
},
"weights": "weights/SpectraVision-1C-CNN.safetensors"
},
"SpectraVision-3C-CNN": {
"encoding": "SpectraVision-3C",
"encoding_params": {
"max_mz": 5000,
"num_peaks": 200,
"bin_multiply_factor": 10
},
"model_params": {
"img_size": [224, 224]
},
"weights": "weights/SpectraVision-3C-CNN.safetensors"
},
"SpectraVision-PeakViT": {
"encoding": "SpectraVision-1C",
"encoding_params": {
"max_mz": 5000,
"num_peaks": 200,
"bin_multiply_factor": 10
},
"model_params": {
"img_size": [224, 224],
"patch_size": 16,
"in_channels": 1,
"d_model": 512,
"embedding_dim": 256,
"num_layers": 2,
"nhead": 8,
"position_encoding": "learned",
"mask_empty_patches": true,
"bin_multiply_factor": 10,
"max_mz": 5000,
"intensity_channel": 0,
"min_mz_wavelength": 0.2,
"max_mz_wavelength": 1000
},
"weights": "weights/SpectraVision-PeakViT.safetensors"
}
}
}
|