File size: 546 Bytes
e875c7d
 
2870e2c
e875c7d
2870e2c
 
 
 
 
e875c7d
 
 
 
 
 
 
 
 
 
2870e2c
 
 
 
 
 
e875c7d
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
{
  "module": "project_nott",
  "version": "v3_sol",
  "paper": "arXiv:2603.14998",
  "training_dataset": "robotflowlabs/anima-thermal-synthetic",
  "pretrained_on": "VIVID++ (71K frames)",
  "finetuned_on": "SOL thermal synthetic (26K COCO + 399 VIVID real)",
  "val_loss": 0.0785277783870697,
  "val_ARE": 0.17465606331825256,
  "parameters": 2048320,
  "input_shape": [
    1,
    1,
    256,
    320
  ],
  "formats": [
    "pth",
    "safetensors",
    "onnx",
    "trt_fp16",
    "trt_fp32"
  ],
  "frozen_bn": true,
  "ddp_world_size": 8
}