SYMBOL INDEX (336 symbols across 40 files)

FILE: deepcore/datasets/cifar10.py
  function CIFAR10 (line 5) | def CIFAR10(data_path):

FILE: deepcore/datasets/cifar100.py
  function CIFAR100 (line 5) | def CIFAR100(data_path):

FILE: deepcore/datasets/fashionmnist.py
  function FashionMNIST (line 4) | def FashionMNIST(data_path):

FILE: deepcore/datasets/imagenet.py
  function ImageNet (line 5) | def ImageNet(data_path):

FILE: deepcore/datasets/mnist.py
  function MNIST (line 5) | def MNIST(data_path, permuted=False, permutation_seed=None):
  function permutedMNIST (line 24) | def permutedMNIST(data_path, permutation_seed=None):

FILE: deepcore/datasets/qmnist.py
  function QMNIST (line 4) | def QMNIST(data_path):

FILE: deepcore/datasets/svhn.py
  function SVHN (line 5) | def SVHN(data_path):

FILE: deepcore/datasets/tinyimagenet.py
  function TinyImageNet (line 7) | def TinyImageNet(data_path, downsize=True):

FILE: deepcore/methods/cal.py
  class Cal (line 11) | class Cal(EarlyTrain):
    method __init__ (line 12) | def __init__(self, dst_train, args, fraction=0.5, random_seed=None, ep...
    method num_classes_mismatch (line 32) | def num_classes_mismatch(self):
    method while_update (line 35) | def while_update(self, outputs, loss, targets, epoch, batch_idx, batch...
    method find_knn (line 40) | def find_knn(self):
    method calc_kl (line 97) | def calc_kl(self, knn, index=None):
    method finish_run (line 123) | def finish_run(self):
    method select (line 137) | def select(self, **kwargs):

FILE: deepcore/methods/contextualdiversity.py
  class ContextualDiversity (line 9) | class ContextualDiversity(kCenterGreedy):
    method __init__ (line 10) | def __init__(self, dst_train, args, fraction=0.5, random_seed=None, ep...
    method _metric (line 15) | def _metric(self, a_output, b_output):
    method construct_matrix (line 22) | def construct_matrix(self, index=None):

FILE: deepcore/methods/coresetmethod.py
  class CoresetMethod (line 1) | class CoresetMethod(object):
    method __init__ (line 2) | def __init__(self, dst_train, args, fraction=0.5, random_seed=None, **...
    method select (line 15) | def select(self, **kwargs):

FILE: deepcore/methods/craig.py
  class Craig (line 9) | class Craig(EarlyTrain):
    method __init__ (line 10) | def __init__(self, dst_train, args, fraction=0.5, random_seed=None, ep...
    method before_train (line 19) | def before_train(self):
    method after_loss (line 22) | def after_loss(self, outputs, loss, targets, batch_inds, epoch):
    method before_epoch (line 25) | def before_epoch(self):
    method after_epoch (line 28) | def after_epoch(self):
    method before_run (line 31) | def before_run(self):
    method num_classes_mismatch (line 34) | def num_classes_mismatch(self):
    method while_update (line 37) | def while_update(self, outputs, loss, targets, epoch, batch_idx, batch...
    method calc_gradient (line 42) | def calc_gradient(self, index=None):
    method calc_weights (line 74) | def calc_weights(self, matrix, result):
    method finish_run (line 81) | def finish_run(self):
    method select (line 120) | def select(self, **kwargs):

FILE: deepcore/methods/deepfool.py
  class DeepFool (line 6) | class DeepFool(EarlyTrain):
    method __init__ (line 7) | def __init__(self, dst_train, args, fraction=0.5, random_seed=None, ep...
    method num_classes_mismatch (line 14) | def num_classes_mismatch(self):
    method while_update (line 17) | def while_update(self, outputs, loss, targets, epoch, batch_idx, batch...
    method finish_run (line 22) | def finish_run(self):
    method deep_fool (line 47) | def deep_fool(self, inputs):
    method select (line 118) | def select(self, **kwargs):

FILE: deepcore/methods/earlytrain.py
  class EarlyTrain (line 10) | class EarlyTrain(CoresetMethod):
    method __init__ (line 15) | def __init__(self, dst_train, args, fraction=0.5, random_seed=None, ep...
    method train (line 61) | def train(self, epoch, list_of_train_idx, **kwargs):
    method run (line 96) | def run(self):
    method test (line 148) | def test(self, epoch):
    method num_classes_mismatch (line 180) | def num_classes_mismatch(self):
    method before_train (line 183) | def before_train(self):
    method after_loss (line 186) | def after_loss(self, outputs, loss, targets, batch_inds, epoch):
    method while_update (line 189) | def while_update(self, outputs, loss, targets, epoch, batch_idx, batch...
    method finish_train (line 192) | def finish_train(self):
    method before_epoch (line 195) | def before_epoch(self):
    method after_epoch (line 198) | def after_epoch(self):
    method before_run (line 201) | def before_run(self):
    method finish_run (line 204) | def finish_run(self):
    method select (line 207) | def select(self, **kwargs):

FILE: deepcore/methods/forgetting.py
  class Forgetting (line 10) | class Forgetting(EarlyTrain):
    method __init__ (line 11) | def __init__(self, dst_train, args, fraction=0.5, random_seed=None, ep...
    method get_hms (line 18) | def get_hms(self, seconds):
    method before_train (line 26) | def before_train(self):
    method after_loss (line 31) | def after_loss(self, outputs, loss, targets, batch_inds, epoch):
    method while_update (line 39) | def while_update(self, outputs, loss, targets, epoch, batch_idx, batch...
    method before_epoch (line 50) | def before_epoch(self):
    method after_epoch (line 53) | def after_epoch(self):
    method before_run (line 58) | def before_run(self):
    method finish_run (line 64) | def finish_run(self):
    method select (line 67) | def select(self, **kwargs):

FILE: deepcore/methods/full.py
  class Full (line 5) | class Full(CoresetMethod):
    method __init__ (line 6) | def __init__(self, dst_train, args, fraction, random_seed, **kwargs):
    method select (line 9) | def select(self, **kwargs):

FILE: deepcore/methods/glister.py
  class Glister (line 8) | class Glister(EarlyTrain):
    method __init__ (line 9) | def __init__(self, dst_train, args, fraction=0.5, random_seed=None, ep...
    method calc_gradient (line 23) | def calc_gradient(self, index=None, val=False, record_val_detail=False):
    method update_val_gradients (line 81) | def update_val_gradients(self, new_selection, selected_for_train):
    method finish_run (line 108) | def finish_run(self):
    method num_classes_mismatch (line 150) | def num_classes_mismatch(self):
    method while_update (line 153) | def while_update(self, outputs, loss, targets, epoch, batch_idx, batch...

FILE: deepcore/methods/gradmatch.py
  class GradMatch (line 11) | class GradMatch(EarlyTrain):
    method __init__ (line 12) | def __init__(self, dst_train, args, fraction=0.5, random_seed=None, ep...
    method num_classes_mismatch (line 18) | def num_classes_mismatch(self):
    method while_update (line 21) | def while_update(self, outputs, loss, targets, epoch, batch_idx, batch...
    method orthogonal_matching_pursuit (line 26) | def orthogonal_matching_pursuit(self, A, b, budget: int, lam: float = ...
    method orthogonal_matching_pursuit_np (line 77) | def orthogonal_matching_pursuit_np(self, A, b, budget: int, lam: float...
    method calc_gradient (line 125) | def calc_gradient(self, index=None, val=False):
    method finish_run (line 158) | def finish_run(self):
    method select (line 210) | def select(self, **kwargs):

FILE: deepcore/methods/grand.py
  class GraNd (line 7) | class GraNd(EarlyTrain):
    method __init__ (line 8) | def __init__(self, dst_train, args, fraction=0.5, random_seed=None, ep...
    method while_update (line 19) | def while_update(self, outputs, loss, targets, epoch, batch_idx, batch...
    method before_run (line 24) | def before_run(self):
    method finish_run (line 28) | def finish_run(self):
    method select (line 57) | def select(self, **kwargs):

FILE: deepcore/methods/herding.py
  class Herding (line 8) | class Herding(EarlyTrain):
    method __init__ (line 9) | def __init__(self, dst_train, args, fraction=0.5, random_seed=None, ep...
    method num_classes_mismatch (line 32) | def num_classes_mismatch(self):
    method while_update (line 35) | def while_update(self, outputs, loss, targets, epoch, batch_idx, batch...
    method construct_matrix (line 40) | def construct_matrix(self, index=None):
    method before_run (line 60) | def before_run(self):
    method herding (line 63) | def herding(self, matrix, budget: int, index=None):
    method finish_run (line 89) | def finish_run(self):
    method select (line 104) | def select(self, **kwargs):

FILE: deepcore/methods/kcentergreedy.py
  function k_center_greedy (line 8) | def k_center_greedy(matrix, budget: int, metric, device, random_seed=Non...
  class kCenterGreedy (line 68) | class kCenterGreedy(EarlyTrain):
    method __init__ (line 69) | def __init__(self, dst_train, args, fraction=0.5, random_seed=None, ep...
    method num_classes_mismatch (line 100) | def num_classes_mismatch(self):
    method while_update (line 103) | def while_update(self, outputs, loss, targets, epoch, batch_idx, batch...
    method old_construct_matrix (line 108) | def old_construct_matrix(self, index=None):
    method construct_matrix (line 129) | def construct_matrix(self, index=None):
    method before_run (line 149) | def before_run(self):
    method finish_run (line 152) | def finish_run(self):
    method select (line 156) | def select(self, **kwargs):

FILE: deepcore/methods/methods_utils/cossim.py
  function cossim_np (line 5) | def cossim_np(v1, v2):
  function cossim_pair_np (line 12) | def cossim_pair_np(v1):
  function cossim (line 20) | def cossim(v1, v2):
  function cossim_pair (line 27) | def cossim_pair(v1):

FILE: deepcore/methods/methods_utils/euclidean.py
  function euclidean_dist (line 5) | def euclidean_dist(x, y):
  function euclidean_dist_pair (line 15) | def euclidean_dist_pair(x):
  function euclidean_dist_np (line 23) | def euclidean_dist_np(x, y):
  function euclidean_dist_pair_np (line 31) | def euclidean_dist_pair_np(x):

FILE: deepcore/methods/methods_utils/submodular_function.py
  class SubmodularFunction (line 4) | class SubmodularFunction(object):
    method __init__ (line 5) | def __init__(self, index, similarity_kernel=None, similarity_matrix=No...
    method _similarity_kernel (line 24) | def _similarity_kernel(self, similarity_kernel):
  class FacilityLocation (line 28) | class FacilityLocation(SubmodularFunction):
    method __init__ (line 29) | def __init__(self, **kwargs):
    method _similarity_kernel (line 39) | def _similarity_kernel(self, similarity_kernel):
    method calc_gain (line 56) | def calc_gain(self, idx_gain, selected, **kwargs):
    method calc_gain_batch (line 60) | def calc_gain_batch(self, idx_gain, selected, **kwargs):
    method update_state (line 70) | def update_state(self, new_selection, total_selected, **kwargs):
  class GraphCut (line 75) | class GraphCut(SubmodularFunction):
    method __init__ (line 76) | def __init__(self, lam: float = 1., **kwargs):
    method _similarity_kernel (line 84) | def _similarity_kernel(self, similarity_kernel):
    method calc_gain (line 103) | def calc_gain(self, idx_gain, selected, **kwargs):
    method update_state (line 109) | def update_state(self, new_selection, total_selected, **kwargs):
  class LogDeterminant (line 113) | class LogDeterminant(SubmodularFunction):
    method __init__ (line 114) | def __init__(self, **kwargs):
    method _similarity_kernel (line 119) | def _similarity_kernel(self, similarity_kernel):
    method calc_gain (line 136) | def calc_gain(self, idx_gain, selected, **kwargs):
    method update_state (line 142) | def update_state(self, new_selection, total_selected, **kwargs):

FILE: deepcore/methods/methods_utils/submodular_optimizer.py
  class optimizer (line 6) | class optimizer(object):
    method __init__ (line 7) | def __init__(self, args, index, budget:int, already_selected=[]):
  class NaiveGreedy (line 19) | class NaiveGreedy(optimizer):
    method __init__ (line 20) | def __init__(self, args, index, budget:int, already_selected=[]):
    method select (line 23) | def select(self, gain_function, update_state=None, **kwargs):
  class LazyGreedy (line 43) | class LazyGreedy(optimizer):
    method __init__ (line 44) | def __init__(self, args, index, budget:int, already_selected=[]):
    method select (line 47) | def select(self, gain_function, update_state=None, **kwargs):
  class StochasticGreedy (line 81) | class StochasticGreedy(optimizer):
    method __init__ (line 82) | def __init__(self, args, index, budget:int, already_selected=[], epsil...
    method select (line 86) | def select(self, gain_function, update_state=None, **kwargs):
  class ApproximateLazyGreedy (line 116) | class ApproximateLazyGreedy(optimizer):
    method __init__ (line 117) | def __init__(self, args, index, budget:int, already_selected=[], beta:...
    method select (line 121) | def select(self, gain_function, update_state=None, **kwargs):

FILE: deepcore/methods/submodular.py
  class Submodular (line 8) | class Submodular(EarlyTrain):
    method __init__ (line 9) | def __init__(self, dst_train, args, fraction=0.5, random_seed=None, ep...
    method before_train (line 21) | def before_train(self):
    method after_loss (line 24) | def after_loss(self, outputs, loss, targets, batch_inds, epoch):
    method before_epoch (line 27) | def before_epoch(self):
    method after_epoch (line 30) | def after_epoch(self):
    method before_run (line 33) | def before_run(self):
    method num_classes_mismatch (line 36) | def num_classes_mismatch(self):
    method while_update (line 39) | def while_update(self, outputs, loss, targets, epoch, batch_idx, batch...
    method calc_gradient (line 44) | def calc_gradient(self, index=None):
    method finish_run (line 80) | def finish_run(self):
    method select (line 118) | def select(self, **kwargs):

FILE: deepcore/methods/uncertainty.py
  class Uncertainty (line 6) | class Uncertainty(EarlyTrain):
    method __init__ (line 7) | def __init__(self, dst_train, args, fraction=0.5, random_seed=None, ep...
    method before_train (line 21) | def before_train(self):
    method after_loss (line 24) | def after_loss(self, outputs, loss, targets, batch_inds, epoch):
    method before_epoch (line 27) | def before_epoch(self):
    method after_epoch (line 30) | def after_epoch(self):
    method before_run (line 33) | def before_run(self):
    method num_classes_mismatch (line 36) | def num_classes_mismatch(self):
    method while_update (line 39) | def while_update(self, outputs, loss, targets, epoch, batch_idx, batch...
    method finish_run (line 44) | def finish_run(self):
    method rank_uncertainty (line 58) | def rank_uncertainty(self, index=None):
    method select (line 87) | def select(self, **kwargs):

FILE: deepcore/methods/uniform.py
  class Uniform (line 5) | class Uniform(CoresetMethod):
    method __init__ (line 6) | def __init__(self, dst_train, args, fraction=0.5, random_seed=None, ba...
    method select_balance (line 12) | def select_balance(self):
    method select_no_balance (line 24) | def select_no_balance(self):
    method select (line 31) | def select(self, **kwargs):

FILE: deepcore/nets/alexnet.py
  class AlexNet_32x32 (line 12) | class AlexNet_32x32(nn.Module):
    method __init__ (line 13) | def __init__(self, channel, num_classes, record_embedding=False, no_gr...
    method get_last_layer (line 35) | def get_last_layer(self):
    method forward (line 38) | def forward(self, x):
  class AlexNet_224x224 (line 47) | class AlexNet_224x224(models.AlexNet):
    method __init__ (line 48) | def __init__(self, channel: int, num_classes: int, record_embedding: b...
    method get_last_layer (line 60) | def get_last_layer(self):
    method forward (line 63) | def forward(self, x: torch.Tensor) -> torch.Tensor:
  function AlexNet (line 72) | def AlexNet(channel: int, num_classes: int, im_size, record_embedding: b...

FILE: deepcore/nets/inceptionv3.py
  class BasicConv2d (line 7) | class BasicConv2d(nn.Module):
    method __init__ (line 9) | def __init__(self, input_channels, output_channels, **kwargs):
    method forward (line 15) | def forward(self, x):
  class InceptionA (line 24) | class InceptionA(nn.Module):
    method __init__ (line 26) | def __init__(self, input_channels, pool_features):
    method forward (line 46) | def forward(self, x):
  class InceptionB (line 67) | class InceptionB(nn.Module):
    method __init__ (line 69) | def __init__(self, input_channels):
    method forward (line 82) | def forward(self, x):
  class InceptionC (line 102) | class InceptionC(nn.Module):
    method __init__ (line 103) | def __init__(self, input_channels, channels_7x7):
    method forward (line 131) | def forward(self, x):
  class InceptionD (line 149) | class InceptionD(nn.Module):
    method __init__ (line 151) | def __init__(self, input_channels):
    method forward (line 168) | def forward(self, x):
  class InceptionE (line 184) | class InceptionE(nn.Module):
    method __init__ (line 185) | def __init__(self, input_channels):
    method forward (line 203) | def forward(self, x):
  class InceptionV3_32x32 (line 239) | class InceptionV3_32x32(nn.Module):
    method __init__ (line 241) | def __init__(self, channel, num_classes, record_embedding=False, no_gr...
    method get_last_layer (line 276) | def get_last_layer(self):
    method forward (line 279) | def forward(self, x):
  class InceptionV3_224x224 (line 331) | class InceptionV3_224x224(inception.Inception3):
    method __init__ (line 332) | def __init__(self, channel: int, num_classes: int, record_embedding: b...
    method get_last_layer (line 340) | def get_last_layer(self):
    method _forward (line 343) | def _forward(self, x):
  function InceptionV3 (line 400) | def InceptionV3(channel: int, num_classes: int, im_size, record_embeddin...

FILE: deepcore/nets/lenet.py
  class LeNet (line 11) | class LeNet(nn.Module):
    method __init__ (line 12) | def __init__(self, channel, num_classes, im_size, record_embedding: bo...
    method get_last_layer (line 32) | def get_last_layer(self):
    method forward (line 35) | def forward(self, x):

FILE: deepcore/nets/mlp.py
  class MLP (line 14) | class MLP(nn.Module):
    method __init__ (line 15) | def __init__(self, channel, num_classes, im_size, record_embedding: bo...
    method get_last_layer (line 27) | def get_last_layer(self):
    method forward (line 30) | def forward(self, x):

FILE: deepcore/nets/mobilenetv3.py
  function _make_divisible (line 15) | def _make_divisible(v, divisor, min_value=None):
  class h_sigmoid (line 31) | class h_sigmoid(nn.Module):
    method __init__ (line 32) | def __init__(self, inplace=True):
    method forward (line 36) | def forward(self, x):
  class h_swish (line 40) | class h_swish(nn.Module):
    method __init__ (line 41) | def __init__(self, inplace=True):
    method forward (line 45) | def forward(self, x):
  class SELayer (line 49) | class SELayer(nn.Module):
    method __init__ (line 50) | def __init__(self, channel, reduction=4):
    method forward (line 60) | def forward(self, x):
  function conv_3x3_bn (line 67) | def conv_3x3_bn(inp, oup, stride, padding=1):
  function conv_1x1_bn (line 75) | def conv_1x1_bn(inp, oup):
  class InvertedResidual (line 83) | class InvertedResidual(nn.Module):
    method __init__ (line 84) | def __init__(self, inp, hidden_dim, oup, kernel_size, stride, use_se, ...
    method forward (line 121) | def forward(self, x):
  class MobileNetV3_32x32 (line 128) | class MobileNetV3_32x32(nn.Module):
    method __init__ (line 129) | def __init__(self, cfgs, mode, channel=3, num_classes=1000, record_emb...
    method forward (line 166) | def forward(self, x):
    method _initialize_weights (line 175) | def _initialize_weights(self):
    method get_last_layer (line 189) | def get_last_layer(self):
  class MobileNetV3_224x224 (line 193) | class MobileNetV3_224x224(mobilenetv3.MobileNetV3):
    method __init__ (line 194) | def __init__(self, inverted_residual_setting, last_channel,
    method get_last_layer (line 207) | def get_last_layer(self):
    method _forward_impl (line 210) | def _forward_impl(self, x: Tensor) -> Tensor:
  function MobileNetV3 (line 219) | def MobileNetV3(arch: str, channel: int, num_classes: int, im_size, reco...
  function MobileNetV3Large (line 295) | def MobileNetV3Large(channel: int, num_classes: int, im_size, record_emb...
  function MobileNetV3Small (line 301) | def MobileNetV3Small(channel: int, num_classes: int, im_size, record_emb...

FILE: deepcore/nets/nets_utils/parallel.py
  class MyDataParallel (line 4) | class MyDataParallel(DataParallel):
    method __getattr__ (line 5) | def __getattr__(self, name):
    method __setattr__ (line 10) | def __setattr__(self, name, value):

FILE: deepcore/nets/nets_utils/recorder.py
  class EmbeddingRecorder (line 4) | class EmbeddingRecorder(nn.Module):
    method __init__ (line 5) | def __init__(self, record_embedding: bool = False):
    method forward (line 9) | def forward(self, x):
    method __enter__ (line 14) | def __enter__(self):
    method __exit__ (line 17) | def __exit__(self, exc_type, exc_val, exc_tb):

FILE: deepcore/nets/resnet.py
  function conv3x3 (line 13) | def conv3x3(in_planes, out_planes, stride=1):
  class BasicBlock (line 17) | class BasicBlock(nn.Module):
    method __init__ (line 20) | def __init__(self, in_planes, planes, stride=1):
    method forward (line 34) | def forward(self, x):
  class Bottleneck (line 42) | class Bottleneck(nn.Module):
    method __init__ (line 45) | def __init__(self, in_planes, planes, stride=1):
    method forward (line 61) | def forward(self, x):
  class ResNet_32x32 (line 70) | class ResNet_32x32(nn.Module):
    method __init__ (line 71) | def __init__(self, block, num_blocks, channel=3, num_classes=10, recor...
    method get_last_layer (line 87) | def get_last_layer(self):
    method _make_layer (line 90) | def _make_layer(self, block, planes, num_blocks, stride):
    method forward (line 98) | def forward(self, x):
  class ResNet_224x224 (line 112) | class ResNet_224x224(resnet.ResNet):
    method __init__ (line 113) | def __init__(self, block, layers, channel: int, num_classes: int, reco...
    method get_last_layer (line 123) | def get_last_layer(self):
    method _forward_impl (line 126) | def _forward_impl(self, x: Tensor) -> Tensor:
  function ResNet (line 147) | def ResNet(arch: str, channel: int, num_classes: int, im_size, record_em...
  function ResNet18 (line 219) | def ResNet18(channel: int, num_classes: int, im_size, record_embedding: ...
  function ResNet34 (line 224) | def ResNet34(channel: int, num_classes: int, im_size, record_embedding: ...
  function ResNet50 (line 229) | def ResNet50(channel: int, num_classes: int, im_size, record_embedding: ...
  function ResNet101 (line 234) | def ResNet101(channel: int, num_classes: int, im_size, record_embedding:...
  function ResNet152 (line 239) | def ResNet152(channel: int, num_classes: int, im_size, record_embedding:...

FILE: deepcore/nets/vgg.py
  class VGG_32x32 (line 18) | class VGG_32x32(nn.Module):
    method __init__ (line 19) | def __init__(self, vgg_name, channel, num_classes, record_embedding=Fa...
    method forward (line 28) | def forward(self, x):
    method get_last_layer (line 36) | def get_last_layer(self):
    method _make_layers (line 39) | def _make_layers(self, cfg):
  class VGG_224x224 (line 54) | class VGG_224x224(vgg.VGG):
    method __init__ (line 55) | def __init__(self, features: nn.Module, channel: int, num_classes: int...
    method get_last_layer (line 67) | def get_last_layer(self):
    method forward (line 70) | def forward(self, x: Tensor) -> Tensor:
  function VGG (line 79) | def VGG(arch: str, channel: int, num_classes: int, im_size, record_embed...
  function VGG11 (line 111) | def VGG11(channel: int, num_classes: int, im_size, record_embedding: boo...
  function VGG13 (line 116) | def VGG13(channel: int, num_classes: int, im_size, record_embedding: boo...
  function VGG16 (line 121) | def VGG16(channel: int, num_classes: int, im_size, record_embedding: boo...
  function VGG19 (line 126) | def VGG19(channel: int, num_classes: int, im_size, record_embedding: boo...

FILE: deepcore/nets/wideresnet.py
  class BasicBlock (line 12) | class BasicBlock(nn.Module):
    method __init__ (line 13) | def __init__(self, in_planes, out_planes, stride, dropRate=0.0):
    method forward (line 28) | def forward(self, x):
  class NetworkBlock (line 40) | class NetworkBlock(nn.Module):
    method __init__ (line 41) | def __init__(self, nb_layers, in_planes, out_planes, block, stride, dr...
    method _make_layer (line 45) | def _make_layer(self, block, in_planes, out_planes, nb_layers, stride,...
    method forward (line 51) | def forward(self, x):
  class WideResNet_32x32 (line 55) | class WideResNet_32x32(nn.Module):
    method __init__ (line 56) | def __init__(self, depth, num_classes, channel=3, widen_factor=1, drop...
    method get_last_layer (line 90) | def get_last_layer(self):
    method forward (line 93) | def forward(self, x):
  function WideResNet (line 106) | def WideResNet(arch: str, channel: int, num_classes: int, im_size, recor...
  function WRN168 (line 159) | def WRN168(channel: int, num_classes: int, im_size, record_embedding: bo...
  function WRN2810 (line 164) | def WRN2810(channel: int, num_classes: int, im_size, record_embedding: b...
  function WRN282 (line 169) | def WRN282(channel: int, num_classes: int, im_size, record_embedding: bo...
  function WRN502 (line 174) | def WRN502(channel: int, num_classes: int, im_size, record_embedding: bo...
  function WRN1012 (line 179) | def WRN1012(channel: int, num_classes: int, im_size, record_embedding: b...

FILE: main.py
  function main (line 13) | def main():

FILE: utils.py
  class WeightedSubset (line 6) | class WeightedSubset(torch.utils.data.Subset):
    method __init__ (line 7) | def __init__(self, dataset, indices, weights) -> None:
    method __getitem__ (line 13) | def __getitem__(self, idx):
  function train (line 19) | def train(train_loader, network, criterion, optimizer, scheduler, epoch,...
  function test (line 72) | def test(test_loader, network, criterion, epoch, args, rec):
  class AverageMeter (line 117) | class AverageMeter(object):
    method __init__ (line 120) | def __init__(self, name, fmt=':f'):
    method reset (line 125) | def reset(self):
    method update (line 131) | def update(self, val, n=1):
    method __str__ (line 137) | def __str__(self):
  function accuracy (line 142) | def accuracy(output, target, topk=(1,)):
  function str_to_bool (line 159) | def str_to_bool(v):
  function save_checkpoint (line 171) | def save_checkpoint(state, path, epoch, prec):
  function init_recorder (line 176) | def init_recorder():
  function record_train_stats (line 190) | def record_train_stats(rec, step, loss, acc, lr):
  function record_test_stats (line 198) | def record_test_stats(rec, step, loss, acc):
  function record_ckpt (line 205) | def record_ckpt(rec, step):
  class DataLoaderX (line 210) | class DataLoaderX(torch.utils.data.DataLoader):
    method __iter__ (line 211) | def __iter__(self):