Views
No views yet
1import torch
2
3# Load checkpoint
4ckpt = torch.load('purebit_best_70mb.pt')
5print(f"Loss: {ckpt['loss']:.4f}")
6print(f"Bits seen: {ckpt['bits']:,}")
7
8# Model architecture in model.pypurebit_best_70mb.pt - Best checkpoint (loss 0.6863)model.py - Model architecturetrain.py - Training scriptinfer.py - Inference script