ICODE-ADC/lammps
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Remove unnecessary comments

Browse Source
This commit is contained in:
rohskopf
2022-09-06 09:08:42 -06:00
parent 341f5cf40d
commit e563eb41c1
1 changed files with 0 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
python/lammps/mliap/pytorch.py
View File

@ -134,9 +134,6 @@ class TorchWrapper(torch.nn.Module):
descriptors = torch.from_numpy(descriptors).to(dtype=self.dtype, device=self.device).requires_grad_(True) descriptors = torch.from_numpy(descriptors).to(dtype=self.dtype, device=self.device).requires_grad_(True)
elems = torch.from_numpy(elems).to(dtype=torch.long, device=self.device) - 1 elems = torch.from_numpy(elems).to(dtype=torch.long, device=self.device) - 1
#print(self.model)
#print("ASDFASDF")
with torch.autograd.enable_grad(): with torch.autograd.enable_grad():
energy_nn = self.model(descriptors, elems) energy_nn = self.model(descriptors, elems)