MCPcopy Create free account

hub / github.com/WonwoongCho/GDWCT / functions

Functions59 in github.com/WonwoongCho/GDWCT

⨍ Functions 59◇ Types & classes 14
↓ 10 callersMethod__init__
(self, input_dim, output_dim, dim, num_block=1, norm='none', n_group=32, activation='relu')
model.py:57
↓ 8 callersMethodl1_criterion
(self, input, target)
run.py:101
↓ 5 callersMethodtrain
(self)
run.py:270
↓ 4 callersFunctionweights_init
(init_type='gaussian')
utils/util.py:46
↓ 3 callersMethodforward
(self, x)
model.py:354
↓ 2 callersMethodcalc_dis_loss
(self, input_fake, input_real)
model.py:361
↓ 2 callersMethodcalc_gen_loss
(self, input_fake)
model.py:379
↓ 2 callersMethodclamping_alpha
(self,G)
run.py:157
↓ 2 callersFunctionget_scheduler
(optimizer, config, iterations=-1)
utils/util.py:36
↓ 2 callersMethodmake_dataset
(self, dir)
data_loader.py:28
↓ 2 callersMethodreg
(self, x_arr)
run.py:104
↓ 2 callersFunctionsave_img
(print_list, name, index, results_dir)
utils/util.py:20
↓ 2 callersMethodset_gpu
(self)
run.py:81
↓ 2 callersMethodupdate_G
(self, x_A, x_B, isTrain=True)
run.py:161
↓ 1 callersMethod_make_net
(self)
model.py:343
↓ 1 callersMethodcoloring
(self)
model.py:81
↓ 1 callersFunctionges_Aonfig
(config)
utils/util.py:32
↓ 1 callersMethodget_block_diagonal_mask
(self, n_member)
run.py:39
↓ 1 callersFunctionget_loader
Build and return a data loader.
data_loader.py:55
↓ 1 callersMethodinit_network
Create a generator and a discriminator.
run.py:51
↓ 1 callersMethodload_pretrained_model
(self, iteration)
run.py:130
↓ 1 callersFunctionmain
()
run.py:349
↓ 1 callersMethodmake_dir
(self)
run.py:47
↓ 1 callersMethodmodel_save
(self, iteration)
run.py:113
↓ 1 callersMethodtest
(self)
run.py:332
↓ 1 callersMethodtest_ready
(self)
run.py:153
↓ 1 callersMethodtrain_ready
(self)
run.py:147
↓ 1 callersMethodupdate_D
(self, x_A, x_B)
run.py:247
↓ 1 callersMethodupdate_learning_rate
(self)
run.py:141
↓ 1 callersMethodwct
style_size torch.Size([1, 766]) mask_size torch.Size([1, 1, 64, 64]) content_size torch.Size([1, 256, 64, 64]) W_
model.py:116
Method__getitem__
(self, index)
data_loader.py:38
Method__init__
(self, config)
run.py:16
Method__init__
(self, dataset, image_dir, transform, mode)
data_loader.py:11
Method__init__
(self, conv_dim=64, repeat_num=4, norm='in', activation='relu')
model.py:8
Method__init__
(self, conv_dim=64, n_group=32, norm='ln', activation='relu')
model.py:31
Method__init__
(self, s_CT, C, G, mask)
model.py:74
Method__init__
(self, n_group, device, input_dim, mlp_dim, bias_dim, mask, w_alpha=0.4)
model.py:104
Method__init__
(self, input_dim, mask, n_group, bias_dim, mlp_dim, repeat_num=4, norm='ln', device=None)
model.py:146
Method__init__
(self, conv_dim=64, repeat_num=8, mask=None, n_group=16, mlp_dim=256, bias_dim=512, content_
model.py:190
Method__init__
(self, dim, norm='in', n_group=32, activation='relu', use_affine=True)
model.py:203
Method__init__
(self, input_dim, output_dim, k, s, p, dilation=False, norm='in', n_group=32, activa
model.py:215
Method__init__
(self, input_dim, output_dim, norm='ln', n_group=32, activation='relu', use_affine=True)
model.py:268
Method__init__
(self, size=None, scale_factor=None, mode='nearest', align_corners=None)
model.py:307
Method__init__
(self, input_dim, params)
model.py:328
Method__len__
(self)
data_loader.py:52
Methodextra_repr
(self)
model.py:317
Methodforward
(self, x)
model.py:26
Methodforward
(self, x)
model.py:52
Methodforward
(self, x)
model.py:69
Methodforward
(self, c_A, s_B)
model.py:113
Methodforward
(self, c_A, s_B)
model.py:167
Methodforward
(self, c_A, s_B_)
model.py:197
Methodforward
(self, x)
model.py:210
Methodforward
(self, x)
model.py:258
Methodforward
(self, x)
model.py:297
Methodforward
(self, input)
model.py:314
Functioninit_fun
(m)
utils/util.py:47
Methodmulti_gpu
(gpu1, gpu2, model)
run.py:82
Functionprint_network
Print out the network information.
utils/util.py:68