JiaruiLi/DeepHealthExpo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix exposure encoding checkpoint loading

Browse Source
This commit is contained in:
Jiarui Li
2026-07-12 09:57:36 +08:00
parent 61cef300ce
commit 8e87d57323
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
encode_exposure_cache.py
View File

@@ -67,7 +67,12 @@ def main() -> None:
if output_path.exists() and not args.overwrite: if output_path.exists() and not args.overwrite:
raise FileExistsError(f"{output_path} exists; pass --overwrite") raise FileExistsError(f"{output_path} exists; pass --overwrite")
checkpoint_data = torch.load(args.checkpoint, map_location="cpu") try:
checkpoint_data = torch.load(
args.checkpoint, map_location="cpu", weights_only=False
)
except TypeError:
checkpoint_data = torch.load(args.checkpoint, map_location="cpu")
model_cfg = checkpoint_data["model_config"] model_cfg = checkpoint_data["model_config"]
normalization = checkpoint_data["normalization"] normalization = checkpoint_data["normalization"]
encoder = TimesNetExposureEncoder( encoder = TimesNetExposureEncoder(