cmustrudel / dire Goto Github PK

(venv) ed@ed-Precision-7540:~/Documents/DIRE/neural-model$ python exp.py test --cuda --extra-config='{"decoder": {"remove_duplicates_in_prediction": true} }' '/media/ed/Dell Portable Hard Drive/dire-data/pretrained_models/data/saved_models/model.hybrid.bin' '/media/ed/Dell Portable Hard Drive/dire-data/preprocessed_splits/test.tar'
Main process id 9942
use random seed 0
loading model from [/media/ed/Dell Portable Hard Drive/dire-data/pretrained_models/data/saved_models/model.hybrid.bin]
{'gnn': {'hidden_size': 128, 'layer_timesteps': [8], 'residual_connections': {'0': [0]}}, 'connections': ['top_down', 'bottom_up', 'terminals', 'variable_master_nodes', 'func_root_to_arg'], 'vocab_file': 'data/all_trees_tokenized_0410/vocab.bpe10000.old_gnn/vocab', 'bpe_model_path': None, 'node_syntax_type_embedding_size': 64, 'node_type_embedding_size': 64, 'node_content_embedding_size': 128, 'init_with_seq_encoding': False, 'decoder_hidden_size': 256, 'dropout': 0.2}
Traceback (most recent call last):
  File "exp.py", line 217, in <module>
    test(cmd_args)
  File "exp.py", line 186, in test
    model = model_cls.load(model_path, use_cuda=args['--cuda'], new_config=extra_config)
  File "/home/ed/Documents/DIRE/neural-model/model/model.py", line 149, in load
    model = cls.build(config, **kwargs)
  File "/home/ed/Documents/DIRE/neural-model/model/model.py", line 64, in build
    encoder = globals()[config['encoder']['type']].build(config['encoder'])
  File "/home/ed/Documents/DIRE/neural-model/model/hybrid_encoder.py", line 51, in build
    return cls(params)
  File "/home/ed/Documents/DIRE/neural-model/model/hybrid_encoder.py", line 23, in __init__
    self.graph_encoder = GraphASTEncoder.build(config['graph_encoder'])
  File "/home/ed/Documents/DIRE/neural-model/model/graph_encoder.py", line 107, in build
    vocab = Vocab.load(params['vocab_file'])
  File "/home/ed/Documents/DIRE/neural-model/utils/vocab.py", line 170, in load
    params = json.load(open(path, 'r'))
FileNotFoundError: [Errno 2] No such file or directory: 'data/all_trees_tokenized_0410/vocab.bpe10000.old_gnn/vocab'

python exp.py \
    train \
    --cuda \
    --work-dir=exp_runs/dire.hybrid \
    --extra-config='{ "data": {"train_file": "data/preprocessed_data/train-shard-*.tar" }, "decoder": { "input_feed": false, "tie_embedding": true }, "train": { "evaluate_every_nepoch": 5, "max_epoch": 60 } }' \
    data/config/model.hybrid.jsonnet