CalvinBackup/SimSwapPlus
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

distillation

Browse Source
This commit is contained in:
chenxuanhong
2022-03-04 19:04:36 +08:00
parent 39964bf613
commit e02f756116
10 changed files with 1232 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files
components/Generator_ori_config.py
+5 -5
View File
@@ -5,7 +5,7 @@
# Created Date: Sunday January 16th 2022
# Author: Chen Xuanhong
# Email: chenxuanhongzju@outlook.com
# Last Modified: Thursday, 3rd March 2022 6:09:43 pm
# Last Modified: Friday, 4th March 2022 1:41:59 am
# Modified By: Chen Xuanhong
# Copyright (c) 2022 Shanghai Jiao Tong University
#############################################################
@@ -106,8 +106,8 @@ class Generator(nn.Module):
activation = nn.ReLU(True)
self.first_layer = nn.Sequential(nn.ReflectionPad2d(3),
nn.Conv2d(3, in_channel, kernel_size=7, padding=0, bias=False),
self.first_layer = nn.Sequential(nn.ReflectionPad2d(1),
nn.Conv2d(3, in_channel, kernel_size=3, padding=0, bias=False),
nn.BatchNorm2d(in_channel), activation)
### downsample
self.down1 = nn.Sequential(nn.Conv2d(in_channel, in_channel*2, kernel_size=3, stride=2, padding=1, bias=False),
@@ -153,8 +153,8 @@ class Generator(nn.Module):
nn.Conv2d(in_channel*2, in_channel, kernel_size=3, stride=1, padding=1, bias=False),
nn.BatchNorm2d(in_channel), activation
)
self.last_layer = nn.Sequential(nn.ReflectionPad2d(3),
nn.Conv2d(in_channel, 3, kernel_size=7, padding=0))
self.last_layer = nn.Sequential(nn.ReflectionPad2d(1),
nn.Conv2d(in_channel, 3, kernel_size=3, padding=0))
# self.__weights_init__()