Views
No views yet
| Metric | Value |
|---|---|
| val/loss | 2.256 |
| val/iou | 0.226 |
| Epochs | 50 |
| Batch size | 44 per GPU |
| GPUs | 2x NVIDIA L4 (DDP) |
| Precision | FP16 mixed |
| Format | File | Size |
|---|---|---|
| PyTorch (.pth) | project_ragnarok_cuda_v1_phase1_best.pth | 128MB |
| SafeTensors | project_ragnarok_cuda_v1_phase1_best.safetensors | 128MB |
| ONNX (opset 17) | project_ragnarok_v1_phase1.onnx | 111MB |
| TensorRT FP32 | project_ragnarok_v1_phase1_trt_fp32.engine | 114MB |
| TensorRT FP16 | project_ragnarok_v1_phase1_trt_fp16.engine | 57MB |
1import torch
2from anima_ragnarok.model import RagnarokModel
3
4model = RagnarokModel(config_path="configs/model_config.yaml")
5state = torch.load("project_ragnarok_cuda_v1_phase1_best.pth")
6model.load_state_dict(state["model_state_dict"], strict=False)
7model.eval()
8
9# Input: 8-channel image [B, 8, 352, 352]
10output = model(input_tensor)
11mask = output[0] # [B, 1, 352, 352]