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

Fix multi_modal_train.py

Browse Source
This commit is contained in:
James Betker 2020-10-26 11:10:22 -06:00
parent f857eb00a8
commit b2f803588b
1 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
codes/multi_modal_train.py
View File

@ -42,14 +42,11 @@ def main(master_opt, launcher):
if __name__ == '__main__':
parser = argparse.ArgumentParser()
#parser.add_argument('-opt', type=str, help='Path to option YAML file.', default='../options/train_exd_imgset_chained_structured_trans_invariance.yml')
parser.add_argument('-opt', type=str, help='Path to option YAML file.', default='../options/train_exd_imgset_chained_structured_trans_invariance.yml')
parser.add_argument('--launcher', choices=['none', 'pytorch'], default='none', help='job launcher')
args = parser.parse_args()
Loader, Dumper = OrderedYaml()
with open(args.opt, mode='r') as f:
opt = yaml.load(f, Loader=Loader)
opt = {
'trainer_options': ['../options/teco.yml', '../options/exd.yml']
}
main(opt, args.launcher)
main(opt, args.launcher)