Skip to content

Commit e9e3b71

Browse files
author
sfluegel
committed
remove inconsistent chebi version, log trainer init args
1 parent 814ba37 commit e9e3b71

File tree

2 files changed

+1
-2
lines changed

2 files changed

+1
-2
lines changed

chebai/trainer/CustomTrainer.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@ def __init__(self, *args, **kwargs):
2020
super().__init__(*args, **kwargs)
2121
# instantiation custom logger connector
2222
self._logger_connector.on_trainer_init(self.logger, 1)
23+
self.logger.log_hyperparams(self.init_kwargs)
2324

2425
def predict_from_file(
2526
self,

configs/data/chebi100.yml

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1 @@
11
class_path: chebai.preprocessing.datasets.chebi.ChEBIOver100
2-
init_args:
3-
chebi_version: 231

0 commit comments

Comments
 (0)