Commit 60a12f60 authored by nd-02110114's avatar nd-02110114
Browse files

🐛 fix gpu bug

parent b504f62a

deepchem/models/torch_models/torch_model.py

+1 −1
Original line number Diff line number Diff line
@@ -179,7 +179,7 @@ class TorchModel(Model): 
      else:

        device = torch.device('cpu')

    self.device = device

    self.model.to(device)

    self.model = model.to(device)



    # W&B logging

    if wandb and not _has_wandb:

CRA Git | Maintained and supported by SUSTech CRA and CCSE