Jens/geointel
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Prepare GeoIntel for public release
Managed validation / Managed repository validation (pull_request) Successful in 1m46s
Details
GeoIntel release gates / Compile, test, contracts and builds (pull_request) Successful in 1m51s
Details
GeoIntel release gates / Python and npm vulnerability policy (pull_request) Successful in 20s
Details
GeoIntel release gates / Production AI image, SBOM and container scan (pull_request) Successful in 15m3s
Details
GeoIntel release gates / Deploy exact gated revision to Unraid (pull_request) Skipped
Details

Browse Source
This commit is contained in:
Jens
2026-08-31 21:33:10 +02:00
parent dcd67b11d1
commit 2cdf9c99c6
195 changed files with 595 additions and 207595 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/train_building_proposal_classifier.py
+2 -2
View File
@@ -319,7 +319,7 @@ def main() -> int:
state_path = args.output_dir / "best-state.pt"
if not state_path.is_file():
raise RuntimeError(f"missing existing best state: {state_path}")
model.load_state_dict(torch.load(state_path, map_location=device))
model.load_state_dict(torch.load(state_path, map_location=device, weights_only=True))
model.eval()
torch.jit.script(model).save(str(args.output_dir / "proposal-classifier.torchscript.pt"))
print(
@@ -369,7 +369,7 @@ def main() -> int:
if float(metric["f1"]) > best_f1:
best_f1 = float(metric["f1"])
torch.save(model.state_dict(), args.output_dir / "best-state.pt")
model.load_state_dict(torch.load(args.output_dir / "best-state.pt", map_location=device))
model.load_state_dict(torch.load(args.output_dir / "best-state.pt", map_location=device, weights_only=True))
model.eval()
scripted = torch.jit.script(model)
model_path = args.output_dir / "proposal-classifier.torchscript.pt"