We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent d019c4c commit 5e7148bCopy full SHA for 5e7148b
1 file changed
libmultilabel/nn/model.py
@@ -199,7 +199,7 @@ def __init__(
199
):
200
super().__init__(num_classes=len(classes), log_path=log_path, **kwargs)
201
self.save_hyperparameters(
202
- ignore=["log_path", "word_dict", "network"]
+ ignore=["log_path"]
203
) # If log_path is saved, loading the checkpoint will cause an error since each experiment has unique log_path (result_dir).
204
self.word_dict = word_dict
205
self.classes = classes
0 commit comments