stockpro-ml / app /models

Commit History

fix: remap bare Sequential keys to net.* when loading ddg_da.pt
ef2e4d9

will702 commited on

fix: patch torchmetrics._apply using CPU probe tensor instead of self.device
ec4688a

will702 commited on

fix: use callable map_location to skip Lightning model.to() — avoids torchmetrics CUDA assertion
49e908a

will702 commited on

fix: patch raw checkpoint to replace all CUDA refs before load
ae54b1b

will702 commited on

fix: patch torchmetrics._apply for CPU-only load of GPU-trained checkpoint
c2a87a5

will702 commited on

fix: track app/models/ package, ignore only top-level /models/
7af21f9

will702 commited on