Tasks that it can do.
=== data Read data from here. It is in gitignore so it won't appear here. === docs Documentation using mkdocs. === image_similarity ====== cluster_images.py Clusters the embeddings learnt using PCA and T-SNE. ====== torch_data.py Contains Dataset class to create PyTorch dataset from folder. ====== torch_model.py Convolutional Auto-enocder Model. ====== torch_engine.py Train_step and validation_step for training. ====== torch_train.py Trainng script. Trains Auto-enocder and saves the embeddings. ====== torch_infer.py Contains inference code ====== config.py Configurations of models and paths. ====== torch_inference.ipynb Inference code in .ipynb to play with. ====== torch_train.ipynb Stand-alone code to train in Jupyter. === tests Contains tests for CI
Don't forget to tag @oke-aditya in your comment, otherwise they may not be notified.