deviandice/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Misc

Browse Source
This commit is contained in:
James Betker 2020-06-18 11:28:55 -06:00
parent 59b0533b06
commit 45a900fafe
3 changed files with 16 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.idea/misc.xml
View File

@ -3,5 +3,4 @@
<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 (python37-torch)" 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 (python37-torch)" jdkType="Python SDK" /> <orderEntry type="jdk" jdkName="Python 3.7 (torch-1.4-venv)" 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>

16
codes/train.py
View File

@ -15,6 +15,7 @@ import options.options as option
from utils import util from utils import util
from data import create_dataloader, create_dataset from data import create_dataloader, create_dataset
from models import create_model from models import create_model
from time import time
def init_dist(backend='nccl', **kwargs): def init_dist(backend='nccl', **kwargs):
@ -30,7 +31,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_imgset_residualgenerator.yml') parser.add_argument('-opt', type=str, help='Path to option YAML file.', default='../options/train_imgset_residualgenerator_fast_specific.yml')
parser.add_argument('--launcher', choices=['none', 'pytorch'], default='none', parser.add_argument('--launcher', choices=['none', 'pytorch'], default='none',
help='job launcher') help='job launcher')
parser.add_argument('--local_rank', type=int, default=0) parser.add_argument('--local_rank', type=int, default=0)
@ -174,7 +175,14 @@ def main():
if opt['dist']: if opt['dist']:
train_sampler.set_epoch(epoch) train_sampler.set_epoch(epoch)
tq_ldr = tqdm(train_loader) tq_ldr = tqdm(train_loader)
_t = time()
_profile = False
for _, train_data in enumerate(tq_ldr): for _, train_data in enumerate(tq_ldr):
if _profile:
print("Data fetch: %f" % (time() - _t))
_t = time()
current_step += 1 current_step += 1
if current_step > total_iters: if current_step > total_iters:
break break
@ -182,8 +190,14 @@ def main():
model.update_learning_rate(current_step, warmup_iter=opt['train']['warmup_iter']) model.update_learning_rate(current_step, warmup_iter=opt['train']['warmup_iter'])
#### training #### training
if _profile:
print("Update LR: %f" % (time() - _t))
_t = time()
model.feed_data(train_data) model.feed_data(train_data)
model.optimize_parameters(current_step) model.optimize_parameters(current_step)
if _profile:
print("Model feed + step: %f" % (time() - _t))
_t = time()
#### log #### log
if current_step % opt['logger']['print_freq'] == 0: if current_step % opt['logger']['print_freq'] == 0: