Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/LetheSec/PLG-MI-Attack
/ functions
Functions
329 in github.com/LetheSec/PLG-MI-Attack
⨍
Functions
329
◇
Types & classes
88
Function
load_attri
(file_path)
baselines/dataloader.py:227
Function
load_image_list
(args, file_path, mode='gan')
baselines/utils.py:453
Function
load_mnist
()
baselines/dataloader.py:177
Function
load_model
Load trained weight. Args: checkpoint_path (str) model (nn.Module) Returns: model
utils.py:229
Function
load_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
Function
load_optim
Load optimizer from checkpoint. Args: checkpoint_path (str) optim (torch.optim) Returns: optim
utils.py:244
Function
load_params
(self, model)
baselines/utils.py:148
Function
load_pretrain
(self, state_dict)
baselines/utils.py:128
Function
load_state_dict
(self, state_dict)
baselines/utils.py:139
Function
max_margin_loss
(out, iden)
losses.py:7
Function
my_softXEnt
(input, target)
baselines/utils.py:431
Function
noise_loss
(V, img1, img2)
baselines/losses.py:12
Function
poincare_loss
(outputs, targets, xi=1e-4)
losses.py:20
Method
predict
(self, x)
baselines/classify.py:57
Method
predict
(self, x)
baselines/classify.py:91
Method
predict
(self, x)
baselines/classify.py:132
Method
predict
(self, x)
models/classifiers/classifier.py:57
Method
predict
(self, x)
models/classifiers/classifier.py:91
Method
predict
(self, x)
models/classifiers/classifier.py:132
Function
print_params
(info, params, dataset=None)
baselines/utils.py:163
Function
resume_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
Function
sample_random_batch
* inputs: - dataset (torch.utils.data.Dataset, required) An instance of torch.utils.data.Dataset. - batch_size (i
baselines/utils.py:295
Function
save_checkpoints
Save checkpoints. Args: args (argparse object) n_iter (int) gen (nn.Module) opt_gen (torch.optim) dis (nn
utils.py:123
Function
save_images
Save images (torch.tensor). Args: root (str) train_image_root (root) fake (torch.tensor) real (torch.tensor)
utils.py:94
Function
train_reg
(args, model, criterion, optimizer, trainloader, testloader, n_epochs)
baselines/engine.py:36
Function
train_reg_at
(args, model, criterion, optimizer, trainloader, testloader, n_epochs)
baselines/engine.py:168
Function
train_vib
(args, model, criterion, optimizer, trainloader, testloader, n_epochs)
baselines/engine.py:86
Function
weights_init
(m)
baselines/utils.py:73
Method
write
(self, data)
baselines/utils.py:36
← previous
301–329 of 329, ranked by callers