/ml_ops_work

Primary LanguagePythonOtherNOASSERTION

MNIST Fashion

To start training run:

python mlops/train_model.py

To make predictions:

python mlops/predict_model.py [CHECKPOINT_PATH] [DATA_PATH]

where DATA_PATH has to be a .pt file containing loaded images.

To visualize the weights in the 4th fully connected layer in a 2d space: To make predictions:

python mlops/visualizations/visualize.py