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"
    }
  }
}