diff --git a/.gitignore b/.gitignore index 50990fe..3b3bf2f 100644 --- a/.gitignore +++ b/.gitignore @@ -148,3 +148,6 @@ _html/ # Project initialization script .initialize_new_project.sh + +# Model files +**/*.pth \ No newline at end of file diff --git a/example_config.toml b/example_config.toml index fa1bb61..768ef00 100644 --- a/example_config.toml +++ b/example_config.toml @@ -41,27 +41,21 @@ timeout = 3600 chunksize = 990 [model] -# name = "ExampleCNN" -# name = "ExampleAutoencoder" - -# An example of requesting an external model class -# external_class = "user_package.submodule.ExternalModel" -external_cls = "kbmod_ml.models.cnn.CNN" +# The name of the built-in model to use or the libpath to an external model +# e.g. "user_package.submodule.ExternalModel" or "ExampleAutoencoder" +name = "kbmod_ml.models.cnn.CNN" weights_filepath = "example_model.pth" epochs = 10 [data_loader] -# Name of data loader to use +# Name of the built-in data loader to use or the libpath to an external data loader +# e.g. "user_package.submodule.ExternalDataLoader" or "HSCDataLoader" name = "CifarDataLoader" # name = "HSCDataLoader" -# An example of requesting an external data loader class -# external_class = "user_package.submodule.ExternalDataLoader" - # Directory path where the data is stored -path = "/Users/drew/code/fibad/data/cifar-10-batches-py" -# path = "/Users/drew/code/fibad/data/hsc-samples" +path = "/home/drew/code/fibad/data/" # Default PyTorch DataLoader parameters batch_size = 10