Code for transferring models to TFLite:
generate_tflite.py - Generates TFLite files
test_tflite.py - Tests generated tflite files
mobile_model.py - Contains models for tflite (Note there is a heavily annotated version of the sparse coding model here. Must be updated if sparse coding is changed.)
Dependencies: tensorflow == 2.6.0
torch == 1.8.0