Merge branch 'checkpoint-disabling' of https://github.com/dice-group/…

…dice-embeddings into checkpoint-disabling

.gitignore

@@ -1,40 +1,6 @@
-dicee/__pycache__/__init__.cpython-310.pyc
-dicee/__pycache__/abstracts.cpython-310.pyc
-dicee/__pycache__/callbacks.cpython-310.pyc
-dicee/__pycache__/dataset_classes.cpython-310.pyc
-dicee/__pycache__/evaluator.cpython-310.pyc
-dicee/__pycache__/executer.cpython-310.pyc
-dicee/__pycache__/knowledge_graph_embeddings.cpython-310.pyc
-dicee/__pycache__/knowledge_graph.cpython-310.pyc
-dicee/__pycache__/query_generator.cpython-310.pyc
-dicee/__pycache__/sanity_checkers.cpython-310.pyc
-dicee/__pycache__/static_funcs_training.cpython-310.pyc
-dicee/__pycache__/static_funcs.cpython-310.pyc
-dicee/__pycache__/static_preprocess_funcs.cpython-310.pyc
-dicee/models/__pycache__/__init__.cpython-310.pyc
-dicee/models/__pycache__/base_model.cpython-310.pyc
-dicee/models/__pycache__/clifford.cpython-310.pyc
-dicee/models/__pycache__/complex.cpython-310.pyc
-dicee/models/__pycache__/dualE.cpython-310.pyc
-dicee/models/__pycache__/function_space.cpython-310.pyc
-dicee/models/__pycache__/octonion.cpython-310.pyc
-dicee/models/__pycache__/pykeen_models.cpython-310.pyc
-dicee/models/__pycache__/quaternion.cpython-310.pyc
-dicee/models/__pycache__/real.cpython-310.pyc
-dicee/models/__pycache__/static_funcs.cpython-310.pyc
-dicee/models/__pycache__/transformers.cpython-310.pyc
-dicee/read_preprocess_save_load_kg/__pycache__/__init__.cpython-310.pyc
-dicee/read_preprocess_save_load_kg/__pycache__/preprocess.cpython-310.pyc
-dicee/read_preprocess_save_load_kg/__pycache__/read_from_disk.cpython-310.pyc
-dicee/read_preprocess_save_load_kg/__pycache__/save_load_disk.cpython-310.pyc
-dicee/read_preprocess_save_load_kg/__pycache__/util.cpython-310.pyc
-dicee/trainer/__pycache__/__init__.cpython-310.pyc
-dicee/trainer/__pycache__/dice_trainer.cpython-310.pyc
-dicee/trainer/__pycache__/torch_trainer_ddp.cpython-310.pyc
-dicee/trainer/__pycache__/torch_trainer.cpython-310.pyc
-dicee.egg-info/dependency_links.txt
-dicee.egg-info/entry_points.txt
-dicee.egg-info/PKG-INFO
-dicee.egg-info/requires.txt
-dicee.egg-info/SOURCES.txt
-dicee.egg-info/top_level.txt
+*.egg-info/
+KGs
+KGs.zip
+checkpoints
+Experiments
+__pycache__/

dicee/scripts/run.py

@@ -123,7 +123,7 @@ def get_default_arguments(description=None):
                         help="Stochastic weight averaging")
     parser.add_argument('--degree', type=int, default=0,
                         help='degree for polynomial embeddings')
-    parser.add_argument('--disable_checkpointing', action='store_false', help='Disable creation of checkpoints during training')
+    parser.add_argument('--disable_checkpointing', action='store_true', help='Disable creation of checkpoints during training')
 
     if description is None:
         return parser.parse_args()

dicee/trainer/dice_trainer.py

@@ -76,7 +76,7 @@ def initialize_trainer(args, callbacks):
                           min_steps=kwargs.get("min_steps", None),
                           detect_anomaly=False,
                           barebones=False,
-                          enable_checkpointing=kwargs['disable_checkpointing'])
+                          enable_checkpointing=not kwargs.get('disable_checkpointing', False))
     else:
         print('Initialize TorchTrainer CPU Trainer', end='\t')
         return TorchTrainer(args, callbacks=callbacks)