[Bug] Fix JIT Could not cast attribute 'rev_species' to type Dict[Tensor, int] - Githubissues

aiqm / torchani

Accurate Neural Network Potential on PyTorch

https://aiqm.github.io/torchani/

MIT License

459 stars 126 forks source link

[Bug] Fix JIT Could not cast attribute 'rev_species' to type Dict[Tensor, int] #596

Closed yueyericardo closed 3 years ago

yueyericardo commented 3 years ago

Credit: codes from: @IgnacioJPickering