issues
search
jpata
/
particleflow
Machine-learned, GPU-accelerated particle flow reconstruction
Apache License 2.0
24
stars
30
forks
source link
Fix checkpoint loading
#280
Closed
erwulff
closed
11 months ago
erwulff
commented
11 months ago
Change
--load-checkpoint
from taking only a checkpoint file name to taking the full path to the checkpoint
Support loading checkpoints from anywhere, not only from within the given training directory
Support checkpoint files with any file extension (i.e. both
.pth
and
.pt
as opposed to only
.pth
like previously)
--load-checkpointfrom taking only a checkpoint file name to taking the full path to the checkpoint.pthand.ptas opposed to only.pthlike previously)