MCPcopy Create free account

hub / github.com/davda54/sam / functions

Functions40 in github.com/davda54/sam

↓ 3 callersMethod__init__
(self, in_channels: int, out_channels: int, stride: int, depth: int, dropout: float)
example/model/wide_res_net.py:47
↓ 3 callersMethodflush
(self)
example/utility/log.py:34
↓ 3 callersFunctionsmooth_crossentropy
(pred, gold, smoothing=0.1)
example/model/smooth_cross_entropy.py:6
↓ 2 callersMethod_reset
(self, len_dataset: int)
example/utility/log.py:82
↓ 2 callersMethod_time
(self)
example/utility/log.py:88
↓ 2 callersMethodeval
(self, len_dataset: int)
example/utility/log.py:23
↓ 2 callersMethodfirst_step
(self, zero_grad=False)
sam.py:16
↓ 2 callersMethodsecond_step
(self, zero_grad=False)
sam.py:30
↓ 2 callersMethodtrain
(self, len_dataset: int)
example/utility/log.py:12
↓ 1 callersMethod_eval_step
(self, loss, accuracy)
example/utility/log.py:77
↓ 1 callersMethod_get_statistics
(self)
example/data/cifar.py:34
↓ 1 callersMethod_grad_norm
(self)
sam.py:49
↓ 1 callersMethod_initialize
(self)
example/model/wide_res_net.py:79
↓ 1 callersMethod_print_header
(self)
example/utility/log.py:92
↓ 1 callersMethod_train_step
(self, model, loss, accuracy, learning_rate: float)
example/utility/log.py:54
↓ 1 callersFunctiondisable_running_stats
(model)
example/utility/bypass_bn.py:6
↓ 1 callersFunctionenable_running_stats
(model)
example/utility/bypass_bn.py:14
↓ 1 callersFunctioninitialize
(args, seed: int)
example/utility/initialize.py:5
↓ 1 callersMethodlr
(self)
example/utility/step_lr.py:20
↓ 1 callersMethodstep
(self, closure=None)
sam.py:41
Method__call__
(self, progress: float)
example/utility/loading_bar.py:6
Method__call__
(self, model, loss, accuracy, learning_rate: float = None)
example/utility/log.py:28
Method__call__
(self, image)
example/utility/cutout.py:10
Method__call__
(self, epoch)
example/utility/step_lr.py:7
Method__init__
(self, params, base_optimizer, rho=0.05, adaptive=False, **kwargs)
sam.py:5
Method__init__
(self, length: int = 40)
example/utility/loading_bar.py:2
Method__init__
(self, log_each: int, initial_epoch=-1)
example/utility/log.py:6
Method__init__
(self, size=16, p=0.5)
example/utility/cutout.py:5
Method__init__
(self, optimizer, learning_rate: float, total_epochs: int)
example/utility/step_lr.py:2
Method__init__
(self, channels: int, dropout: float)
example/model/wide_res_net.py:9
Method__init__
(self, in_channels: int, out_channels: int, stride: int, dropout: float)
example/model/wide_res_net.py:26
Method__init__
(self, depth: int, width_factor: int, dropout: float, in_channels: int, labels: int)
example/model/wide_res_net.py:59
Method__init__
(self, batch_size, threads)
example/data/cifar.py:10
Function_disable
(module)
example/utility/bypass_bn.py:7
Function_enable
(module)
example/utility/bypass_bn.py:15
Methodforward
(self, x)
example/model/wide_res_net.py:21
Methodforward
(self, x)
example/model/wide_res_net.py:41
Methodforward
(self, x)
example/model/wide_res_net.py:54
Methodforward
(self, x)
example/model/wide_res_net.py:92
Methodload_state_dict
(self, state_dict)
sam.py:61