search

wengong-jin / hgraph2graph

Hierarchical Generation of Molecular Graphs using Structural Motifs
MIT License
367 stars 108 forks source link

Unrecognized arguments: --nsamples 1000 #36

Open gtteixeira opened 2 years ago

gtteixeira commented 2 years ago

Hi

After run the example: python generate.py --vocab data/chembl/vocab.txt --model ckpt/chembl-pretrained/model.ckpt --nsamples 1000

I got the error when passing the argument --nsamples

_usage: generate.py [-h] --vocab VOCAB [--atom_vocab ATOM_VOCAB] --model MODEL [--seed SEED] [--nsample NSAMPLE] [--rnn_type RNN_TYPE] [--hidden_size HIDDEN_SIZE] [--embed_size EMBED_SIZE] [--batch_size BATCH_SIZE] [--latent_size LATENT_SIZE] [--depthT DEPTHT] [--depthG DEPTHG] [--diterT DITERT] [--diterG DITERG] [--dropout DROPOUT] generate.py: error: unrecognized arguments: --nsamples 1000_

When I remove this arguments everything is ok.

gtteixeira commented 2 years ago

Hi, I have already identified the error, which is in the --nsamples flag. The correct way is to write --nsample. I just have made a pull request fixing this in the README file. thanks.