↓ 1 callersFunctiontrain_one_epoch(
epoch, model, loader, optimizer, loss_fn, args,
lr_scheduler=None, saver=None, output_dir=No
cifar10-100/train.py:635
↓ 1 callersFunctiontrain_one_epoch(model, criterion, optimizer, data_loader, device, epoch, print_freq, scaler=None, T_train=None, aug=None, tri
event/train.py:210
Method__init__(self, in_channels=2, out_channels=256, kernel_size = 3, stride = 1, padding = 1, shortcut= False)
imagenet/embedding_hub.py:33
Method__init__(self, in_channels=2, out_channels=256, kernel_size = 3, stride = 1, padding = 1, shortcut= False)
imagenet/embedding_hub.py:62
Method__init__(self, in_channels=2, num_classes=11,
embed_dims=384, mlp_ratios=4, drop_rate=0.,
cifar10-100/max_former.py:13
Method__init__(self, in_channels=2, out_channels=256, kernel_size = 3, stride = 1, padding = 1, shortcut= False)
cifar10-100/embedding_hub.py:32
Method__init__(self, in_channels=2, out_channels=256, kernel_size = 3, stride = 1, padding = 1, shortcut= False)
cifar10-100/embedding_hub.py:60
Method__init__(self, in_channels=2, num_classes=11,
embed_dims=384, mlp_ratios=4, drop_rate=0.,
cifar10-100/ms_qkformer.py:13
Method__init__(self, inplanes, planes, stride=1, downsample=None, groups=1,
base_width=64, dilation=1)
cifar10-100/max_resnet.py:24
Method__init__(self, inplanes, planes, stride=1, downsample=None, groups=1,
base_width=64, dilation=1)
cifar10-100/max_resnet.py:64