AncTe/ginka-generator
1
0
Fork 0
mirror of https://github.com/unanmed/ginka-generator.git synced 2026-05-24 05:01:41 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: 损失值权重

Browse Source
This commit is contained in:
unanmed 2025-03-25 21:31:56 +08:00
parent 8872de3f13
commit def3cc5c3f
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
minamo/model/loss.py
View File

@ -1,7 +1,7 @@
import torch.nn as nn import torch.nn as nn
class MinamoLoss(nn.Module): class MinamoLoss(nn.Module):
def __init__(self, vision_weight=1, topo_weight=0): def __init__(self, vision_weight=0.4, topo_weight=0.6):
super().__init__() super().__init__()
self.vision_weight = vision_weight self.vision_weight = vision_weight
self.topo_weight = topo_weight self.topo_weight = topo_weight

2
minamo/model/topo.py
View File

@ -1,7 +1,7 @@
import torch import torch
import torch.nn as nn import torch.nn as nn
import torch.nn.functional as F import torch.nn.functional as F
from torch_geometric.nn import GATConv, AttentionalAggregation, global_max_pool from torch_geometric.nn import GATConv, global_max_pool
from torch_geometric.data import Data from torch_geometric.data import Data
class MinamoTopoModel(nn.Module): class MinamoTopoModel(nn.Module):