From 426584582acbb491ac19329111d64e8ba1c588c7 Mon Sep 17 00:00:00 2001 From: Anton Makoveev Date: Tue, 17 Sep 2024 15:35:01 +0200 Subject: [PATCH] [Fix]: remove debug files --- test_models.py | 27 --------------------------- 1 file changed, 27 deletions(-) delete mode 100644 test_models.py diff --git a/test_models.py b/test_models.py deleted file mode 100644 index 31a0634e..00000000 --- a/test_models.py +++ /dev/null @@ -1,27 +0,0 @@ -from pathlib import Path - -from luxonis_train.core import LuxonisModel - -TEST_OUTPUT = Path("./probe") -OPTS = { - "trainer.epochs": 1, - "trainer.batch_size": 4, - "trainer.validation_interval": 1, - "trainer.callbacks": "[]", - "tracker.save_directory": str(TEST_OUTPUT), - "tuner.n_trials": 4, -} - - -def main(): - config_file = "obb_detection_model" - # config_file = "detection_model" - config_file = f"configs/{config_file}.yaml" - # model = LuxonisModel(config_file, opts=OPTS) - model = LuxonisModel(config_file) - model.train() - model.test() - - -if __name__ == "__main__": - main()