↓ 2 callersFunctionfourD2threeD Convert a batch of images (N,W,H,C) to a single big image (W*n, H*m, C) Input: batch: type=ndarray, shape=(N,W,H,C) Return: rows: type=ndarray
utils/utils.py:295
↓ 1 callersFunctionload_ckpt(netG, netD,
optimizer_G, optimizer_D, optimizer_gamma,
lr_scheduler_G, lr_scheduler_D, lr_scheduler_gamma
utils/utils.py:68
↓ 1 callersFunctionsave_ckpt(epoch, netG, netD,
optimizer_G, optimizer_D, optimizer_gamma,
lr_scheduler_G, lr_scheduler_D, lr_schedule
utils/utils.py:44
↓ 1 callersFunctionsave_ckpt_finetune(epoch, netG, netD,
optimizer_G, optimizer_D,
lr_scheduler_G, lr_scheduler_D,
loss_G_lst, loss_G_perceptu
utils/utils.py:96
Method__init__(self, in_channels, out_channels, kernel_size, stride=1,
padding=0, dilation=1, groups=1,
models/models.py:44