include requirements.txt and update readme

mrusci / training-mixed-precision-quantized-networks

This repository containts the pytorch scripts to train mixed-precision networks for microcontroller deployment, based on the memory contraints of the target device.

49 stars 16 forks source link

include requirements.txt and update readme #3

Open Ahmed-Mujtaba-98 opened 6 months ago

Ahmed-Mujtaba-98 commented 6 months ago

Solve the environment setup to start the training. Pytorch 0.4.1 and Python3.5 had a lot of dependency issues with the tensorflow itself.

What's Changed?

Contains a requirements.txt file with all the required libraries.
Updated readme to make the environment seamlessly with conda.

Thanks!