Skip to content

TorchMD-NET. This is a simple code to train Schnet using pytorch-lighthing.

License

Notifications You must be signed in to change notification settings

michaellili/torchmd-net

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

torchmd-net

Installation

conda create -n torchmdnet
conda activate torchmdnet
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
conda install pyyaml ipython scikit-learn tqdm
pip install pytorch-lightning
pip install torchmdnet

About

TorchMD-NET. This is a simple code to train Schnet using pytorch-lighthing.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 99.2%
  • Makefile 0.8%