From 20a4e2d403f2efdc02e254b7306da4d4547b53a5 Mon Sep 17 00:00:00 2001 From: LielinJiang Date: Mon, 28 Sep 2020 12:28:26 +0000 Subject: [PATCH] add init module --- configs/pix2pix_cityscapes.yaml | 2 +- ppgan/modules/init.py | 12 ------------ 2 files changed, 1 insertion(+), 13 deletions(-) diff --git a/configs/pix2pix_cityscapes.yaml b/configs/pix2pix_cityscapes.yaml index 6fe0056..5919ff2 100644 --- a/configs/pix2pix_cityscapes.yaml +++ b/configs/pix2pix_cityscapes.yaml @@ -25,7 +25,7 @@ dataset: train: name: PairedDataset dataroot: data/cityscapes - num_workers: 0 + num_workers: 4 phase: train max_dataset_size: inf direction: BtoA diff --git a/ppgan/modules/init.py b/ppgan/modules/init.py index 1bdb1a9..37e4257 100644 --- a/ppgan/modules/init.py +++ b/ppgan/modules/init.py @@ -298,15 +298,3 @@ def init_weights(net, init_type='normal', init_gain=0.02): print('initialize network with %s' % init_type) net.apply(init_func) # apply the initialization function - - -# if __name__ == "__main__": -# paddle.disable_static() -# conv1 = paddle.nn.Conv2d(3, 8, 1) -# print('bias before', conv1.bias) -# constant_init(conv1, -1., -1.) -# print('bias before', conv1.bias) -# kaiming_init(conv1) -# print('bias after kaiming', conv1.bias) -# xavier_init(conv1) -# print('bias after xavier', conv1.bias) -- GitLab