Views
No views yet
E_omega(T, d, c)T, continuous duty cycle d,
and target specification c to a scalar energy. Lower energy is intended to
correspond to better topology-duty-spec combinations.shared_energy.seal_models.GTNEnergyLossNet class and the checkpoint's
model_state_dict.seal_lossnet_model.pt: PyTorch checkpoint with model_state_dict,
model_config, training config, and best validation metadata.manifest.json: Training and evaluation summary.training_history.jsonl: Per-epoch training and validation metrics, included
when uploaded with --include-eval-artifacts.L_E = L_reg + lambda_rank * (rho * L_intra + (1 - rho) * L_inter)lambda_rank=0.2 and rho=0.5.1182001{
2 "dense_test": 11988,
3 "dense_train": 55809,
4 "dense_val": 11988,
5 "inter_test": 5000,
6 "inter_train": 30000,
7 "inter_val": 5000,
8 "intra_test": 4736,
9 "intra_train": 22048,
10 "intra_val": 4736
11}1{
2 "dense_count": 11988.0,
3 "inter_accuracy": 0.7376,
4 "inter_count": 5000.0,
5 "inter_loss": 0.011845157063007355,
6 "intra_accuracy": 0.7048141891891891,
7 "intra_count": 4736.0,
8 "intra_loss": 0.00825060364110647,
9 "reg_huber": 0.00027953632127988956,
10 "reg_mae": 0.012845741039838718,
11 "total_proxy": 0.0022891123916912726
12}1{
2 "dense_count": 11988.0,
3 "inter_accuracy": 0.7554,
4 "inter_count": 5000.0,
5 "inter_loss": 0.014154602885246276,
6 "intra_accuracy": 0.7185388513513513,
7 "intra_count": 4736.0,
8 "intra_loss": 0.009359711932169425,
9 "reg_huber": 0.00067927523821364,
10 "reg_mae": 0.015292200926426533,
11 "total_proxy": 0.0030307067199552102
12}1from huggingface_hub import hf_hub_download
2import torch
3
4from shared_energy.seal_models import GTNEnergyLossNet
5
6repo_id = "tjwjdgns011119/seal-lossnet-v0"
7checkpoint_path = hf_hub_download(repo_id=repo_id, filename="seal_lossnet_model.pt")
8
9payload = torch.load(checkpoint_path, map_location="cpu")
10model = GTNEnergyLossNet(**payload["model_config"])
11model.load_state_dict(payload["model_state_dict"])
12model.eval()
13
14# energy = model(
15# node_features=node_features,
16# neighbor_mask=neighbor_mask,
17# loop_membership=loop_membership,
18# node_mask=node_mask,
19# duty=duty,
20# spec=spec,
21# )pip install -e . from the repository root.