Loading deepchem/models/keras_model.py +3 −0 Original line number Diff line number Diff line Loading @@ -175,6 +175,9 @@ class KerasModel(Model): the Weights & Biases logger object used to log data and metrics """ super(KerasModel, self).__init__(model=model, model_dir=model_dir, **kwargs) self.loss = loss # not used self.learning_rate = learning_rate # not used self.output_types = output_types # not used if isinstance(loss, Loss): self._loss_fn: LossFn = _StandardLoss(model, loss) else: Loading deepchem/models/torch_models/torch_model.py +3 −0 Original line number Diff line number Diff line Loading @@ -175,6 +175,9 @@ class TorchModel(Model): the Weights & Biases logger object used to log data and metrics """ super(TorchModel, self).__init__(model=model, model_dir=model_dir, **kwargs) self.loss = loss # not used self.learning_rate = learning_rate # not used self.output_types = output_types # not used if isinstance(loss, Loss): self._loss_fn: LossFn = _StandardLoss(self, loss) else: Loading Loading
deepchem/models/keras_model.py +3 −0 Original line number Diff line number Diff line Loading @@ -175,6 +175,9 @@ class KerasModel(Model): the Weights & Biases logger object used to log data and metrics """ super(KerasModel, self).__init__(model=model, model_dir=model_dir, **kwargs) self.loss = loss # not used self.learning_rate = learning_rate # not used self.output_types = output_types # not used if isinstance(loss, Loss): self._loss_fn: LossFn = _StandardLoss(model, loss) else: Loading
deepchem/models/torch_models/torch_model.py +3 −0 Original line number Diff line number Diff line Loading @@ -175,6 +175,9 @@ class TorchModel(Model): the Weights & Biases logger object used to log data and metrics """ super(TorchModel, self).__init__(model=model, model_dir=model_dir, **kwargs) self.loss = loss # not used self.learning_rate = learning_rate # not used self.output_types = output_types # not used if isinstance(loss, Loss): self._loss_fn: LossFn = _StandardLoss(self, loss) else: Loading
