Hi developers,
When I'm trying to train with schnet architecture model, I'm getting the following error. When I used graph-network I received the same error. How can I train schnet architecture using torchmd-net?
torchmd-train --conf t.yaml
Seed set to 1
Number of files: 1
Combined dataset size 1868861
Preloading Custom dataset (of size 427.75 MB)
train 1618159, val 93443, test 157259
Traceback (most recent call last):
File "/home/chemistry/phd/cyz218385/mambaforge/envs/torchmd-net/bin/torchmd-train", line 11, in
sys.exit(main())
File "/home/chemistry/phd/cyz218385/mambaforge/envs/torchmd-net/lib/python3.10/site-packages/torchmdnet/scripts/train.py", line 202, in main
model = LNNP(args, prior_model=prior_models, mean=data.mean, std=data.std)
File "/home/chemistry/phd/cyz218385/mambaforge/envs/torchmd-net/lib/python3.10/site-packages/torchmdnet/module.py", line 89, in init
self.model = create_model(self.hparams, prior_model, mean, std)
File "/home/chemistry/phd/cyz218385/mambaforge/envs/torchmd-net/lib/python3.10/site-packages/torchmdnet/models/model.py", line 110, in create_model
raise ValueError(f'Unknown architecture: {args["model"]}')
ValueError: Unknown architecture: torchmd_gn
Here is my input file
activation: tanh
attn_activation: silu
dataset: Custom
coord_files: "coords.npy"
force_files: "deltaforces.npy"
embed_files: "embeddings.npy"
log_dir: /scratch/chemistry/phd/cyz218385/chig_cgschnet
num_epochs: 200
batch_size: 256
model: torchmd_gn
lr: 0.0001
lr_patience: 4
lr_min: 1.e-06
lr_factor: 0.8
ngpus: 2
num_nodes: 1
num_workers: 8
seed: 1
lr_patience: 1
save_interval: 1
val_size: 0.05
test_size: 157259
#hidden_channels: 128
#num_filters: 128
#num_gaussians: 128
#num_interactions: 3
#max_z: 100
#cutoff: 9
torchmd
