MAC (Model Agnostic Compressor)

This is a starter repo providing skeleton of ML projects for image classification, image segmentation and image generation. Please fork this repo to get started and share your brilliant ideas through pull request.

Quick start

Train and save a model

python main.py train --exp_name="ED0"

Profile the model

MatroidModel python run.py profile --n=100
EfficientDet python run.py profile --n=100 --gpu --network="efficientdet-d7

Calculate the mAP

python run.py detect --gpu
cd Object-Detection-Metrics && python pascalvoc.py

Useful links

Negative sampling (Google Map API)
Example of dataset implementation: USGS dataset
Documentation for Fire
Documentation for Pytorch Lighning

Troubleshooting Notes

Inplace operations in PyTorch is not supported in PyTorch Lightning distributed mode. Please just use non-inplace operations instead.

Maintainers: @Hao