The Honda Research institute data that we used in the project (pre-processed by NeurIPS paper) can be found here
- To pre-train the model using simCLR: Run the Pretrain.ipynb Jupyter Notebook file
- To fine-tune the model: Go to DM_Project/simCLR/fine_tune/ directory and run
python3 main.py
In order to run the autoencoder code, some steps need to be completed first:
- Download the .pt files from this link: https://drive.google.com/drive/folders/1w7GfDKQ_wsClzP8fkrbpOe-Q4nRMjGaj?usp=sharing. These files go in the saved_models folder within the autoencoder folder.
- Download the data from this link: https://drive.google.com/drive/folders/1jzoYxJuIaClt9Q_tloYxK06tuAJrYAFi?usp=sharing. These files go in the data folder within the autoencoder folder.
- Now, you can either open the autoencoder folder within the simCLR folder with VS Code and run the main.py file or you can navigate to the folder through the terminal and run the following command:
python3 main.py
- This will run the code for the autoencoder portion of the project.
To test a trained model on validation sets, run
python3 fine_tune_model.py