mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Use local checkpoint in SSG

Browse Source
This commit is contained in:
James Betker 2020-10-08 08:54:46 -06:00
parent c93dd623d7
commit 969bcd9021
3 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.idea/mmsr.iml
View File

@ -11,7 +11,6 @@
</content>
<orderEntry type="jdk" jdkName="torch-cu11" jdkType="Python SDK" />
<orderEntry type="sourceFolder" forTests="false" />
<orderEntry type="module" module-name="SwitchedConvolutions" />
</component>
<component name="PyDocumentationSettings">
<option name="format" value="PLAIN" />

1
.idea/modules.xml
View File

@ -2,7 +2,6 @@
<project version="4">
<component name="ProjectModuleManager">
<modules>
<module fileurl="file://$PROJECT_DIR$/../SwitchedConvolutions/.idea/SwitchedConvolutions.iml" filepath="$PROJECT_DIR$/../SwitchedConvolutions/.idea/SwitchedConvolutions.iml" />
<module fileurl="file://$PROJECT_DIR$/.idea/mmsr.iml" filepath="$PROJECT_DIR$/.idea/mmsr.iml" />
</modules>
</component>

2
codes/models/archs/StructuredSwitchedGenerator.py
View File

@ -10,7 +10,7 @@ from switched_conv.switched_conv_util import save_attention_to_image_rgb
from switched_conv.switched_conv import compute_attention_specificity
import os
import torchvision
from torch.utils.checkpoint import checkpoint
from utils.util import checkpoint
# VGG-style layer with Conv(stride2)->BN->Activation->Conv->BN->Activation
# Doubles the input filter count.