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