MCPcopy Create free account

hub / github.com/LetheSec/PLG-MI-Attack / functions

Functions329 in github.com/LetheSec/PLG-MI-Attack

Functionload_attri
(file_path)
baselines/dataloader.py:227
Functionload_image_list
(args, file_path, mode='gan')
baselines/utils.py:453
Functionload_mnist
()
baselines/dataloader.py:177
Functionload_model
Load trained weight. Args: checkpoint_path (str) model (nn.Module) Returns: model
utils.py:229
Functionload_module_state_dict
Copies parameters and buffers from :attr:`state_dict` into this module and its descendants. If :attr:`strict` is ``True`` then the keys of :at
baselines/utils.py:192
Functionload_optim
Load optimizer from checkpoint. Args: checkpoint_path (str) optim (torch.optim) Returns: optim
utils.py:244
Functionload_params
(self, model)
baselines/utils.py:148
Functionload_pretrain
(self, state_dict)
baselines/utils.py:128
Functionload_state_dict
(self, state_dict)
baselines/utils.py:139
Functionmax_margin_loss
(out, iden)
losses.py:7
Functionmy_softXEnt
(input, target)
baselines/utils.py:431
Functionnoise_loss
(V, img1, img2)
baselines/losses.py:12
Functionpoincare_loss
(outputs, targets, xi=1e-4)
losses.py:20
Methodpredict
(self, x)
baselines/classify.py:57
Methodpredict
(self, x)
baselines/classify.py:91
Methodpredict
(self, x)
baselines/classify.py:132
Methodpredict
(self, x)
models/classifiers/classifier.py:57
Methodpredict
(self, x)
models/classifiers/classifier.py:91
Methodpredict
(self, x)
models/classifiers/classifier.py:132
Functionprint_params
(info, params, dataset=None)
baselines/utils.py:163
Functionresume_from_args
Load generator & discriminator with their optimizers from args.json. Args: args_path (str): Path to args.json gen_ckpt_path (str)
utils.py:155
Functionsample_random_batch
* inputs: - dataset (torch.utils.data.Dataset, required) An instance of torch.utils.data.Dataset. - batch_size (i
baselines/utils.py:295
Functionsave_checkpoints
Save checkpoints. Args: args (argparse object) n_iter (int) gen (nn.Module) opt_gen (torch.optim) dis (nn
utils.py:123
Functionsave_images
Save images (torch.tensor). Args: root (str) train_image_root (root) fake (torch.tensor) real (torch.tensor)
utils.py:94
Functiontrain_reg
(args, model, criterion, optimizer, trainloader, testloader, n_epochs)
baselines/engine.py:36
Functiontrain_reg_at
(args, model, criterion, optimizer, trainloader, testloader, n_epochs)
baselines/engine.py:168
Functiontrain_vib
(args, model, criterion, optimizer, trainloader, testloader, n_epochs)
baselines/engine.py:86
Functionweights_init
(m)
baselines/utils.py:73
Methodwrite
(self, data)
baselines/utils.py:36
← previous301–329 of 329, ranked by callers