↓ 1 callersFunctiontrain_model(
trainLoader, model, epoch, epoch_number, optimizer, combiner, criterion, cfg, logger, rank=0, use_ap
lib/core/function.py:23
↓ 1 callersFunctionvalid_model(
dataLoader, epoch_number, model, cfg, criterion, logger, device, rank, distributed, **kwargs
)
lib/core/function.py:104
↓ 1 callersFunctionweighted_focal_l1_loss(inputs, targets, weights=None, activate='sigmoid', beta=.2, gamma=1)
lib/loss/loss_impl/focal_R.py:46
↓ 1 callersFunctionweighted_focal_mse_loss(inputs, targets, weights=None, activate='sigmoid', beta=.2, gamma=1)
lib/loss/loss_impl/focal_R.py:36
MethodFDS(self, model, criterion, data, label, meta, meta_data, meta_label, epoch, training, **kwargs)
lib/core/combiner.py:305
Method__init__(self, num_node_emb_list, num_edge_emb_list,
num_layers=5, emb_dim=300, JK='last', dropout=0.
lib/backbone/graph.py:375