↓ 1 callersMethodspn_block(self, X, l, u, Gl, Gm, Gr, D=None, spn_module=None)
generation/gspn.py:212
↓ 1 callersMethodspn_block(self, X, l, u, Gl, Gm, Gr, D=None, spn_module=None)
classification/models/gspn.py:123
↓ 1 callersFunctiontrain_one_epoch(config, model, criterion, data_loader, optimizer, epoch, mixup_fn, lr_scheduler, loss_scaler, model_ema=None,
classification/main.py:208
Method__call__(self, loss, optimizer, clip_grad=None, parameters=None, create_graph=False, update_grad=True)
classification/utils/utils.py:172
Method__init__(
self,
*,
betas,
model_mean_type,
model_var_type,
loss_type
generation/diffusion/gaussian_diffusion.py:159
Method__init__(self, optimizer: torch.optim.Optimizer, milestones, gamma=0.1, warmup_t=0, warmup_lr_init=0, t_in_epochs=True
classification/utils/lr_scheduler.py:123