File size: 948 Bytes
9ddb370
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
# Reproducibility Notes

## Checkpoint Conversion

The released files were converted from training checkpoints with:

```python
ckpt = torch.load("final.pt", map_location="cpu")
state = {k.removeprefix("module."): v.detach().cpu().contiguous() for k, v in ckpt["model"].items()}
safetensors.torch.save_file(state, "model.safetensors")
```

The original `.pt` files contain optimizer and training state. The released
`.safetensors` files contain only the model state dict required for inference.

## Verification

Each converted checkpoint was reloaded with `safetensors.torch.load_file`, and
the tensor key set was checked against the source PyTorch checkpoint. A forward
pass sanity check was also run using the TeX-1500 codebase and a synthetic
`[1, 64, 64, 64]` HSI tensor.

## Data Access

The TeX-1500 dataset is gated on Hugging Face. Users need to log in and accept
the dataset access conditions before downloading samples for reproduction.