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

Misc

Browse Source
This commit is contained in:
James Betker 2020-10-03 16:15:42 -06:00
parent 9b4ed82093
commit 3cbb9ecd45
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.idea/misc.xml
View File

@ -3,5 +3,5 @@
<component name="JavaScriptSettings"> <component name="JavaScriptSettings">
<option name="languageLevel" value="ES6" /> <option name="languageLevel" value="ES6" />
</component> </component>
<component name="ProjectRootManager" version="2" project-jdk-name="Python 3.7 (pytorch-1.7)" project-jdk-type="Python SDK" /> <component name="ProjectRootManager" version="2" project-jdk-name="torch-cu11" project-jdk-type="Python SDK" />
</project> </project>

2
.idea/mmsr.iml
View File

@ -9,7 +9,7 @@
<excludeFolder url="file://$MODULE_DIR$/results" /> <excludeFolder url="file://$MODULE_DIR$/results" />
<excludeFolder url="file://$MODULE_DIR$/tb_logger" /> <excludeFolder url="file://$MODULE_DIR$/tb_logger" />
</content> </content>
<orderEntry type="jdk" jdkName="Python 3.7 (pytorch-1.7)" jdkType="Python SDK" /> <orderEntry type="jdk" jdkName="torch-cu11" jdkType="Python SDK" />
<orderEntry type="sourceFolder" forTests="false" /> <orderEntry type="sourceFolder" forTests="false" />
<orderEntry type="module" module-name="SwitchedConvolutions" /> <orderEntry type="module" module-name="SwitchedConvolutions" />
</component> </component>

2
codes/train.py
View File

@ -32,7 +32,7 @@ def init_dist(backend='nccl', **kwargs):
def main(): def main():
#### options #### options
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument('-opt', type=str, help='Path to option YAML file.', default='../options/train_exd_imgset_ssgr.yml') parser.add_argument('-opt', type=str, help='Path to option YAML file.', default='../options/train_exd_imgset_ssgr_constrained_gan.yml')
parser.add_argument('--launcher', choices=['none', 'pytorch'], default='none', help='job launcher') parser.add_argument('--launcher', choices=['none', 'pytorch'], default='none', help='job launcher')
parser.add_argument('--local_rank', type=int, default=0) parser.add_argument('--local_rank', type=int, default=0)
args = parser.parse_args() args = parser.parse_args()