SkyID
/
deepke
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

update

This commit is contained in:
leo 2019-09-16 20:25:43 +08:00
parent a272420147
commit a3371d7c7e
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
images/GCN.png Normal file
View File

Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 292 KiB

4
main.py
View File

@ -24,9 +24,9 @@ __Models__ = {
}
parser = argparse.ArgumentParser(description='choose your model')
parser.add_argument('--model', type=str, help='model name')
parser.add_argument('--model_name', type=str, help='model name: [CNN, RNN, GCN, Capsule, Transformer, LM]')
args = parser.parse_args()
model_name = args.model if args.model else config.model_name
model_name = args.model_name if args.model_name else config.model_name
make_seed(config.training.seed)