News_Image_Classification Required Dependencies Cuda Python Pytorch Step to train model python Train.py --cuda Step to test model python Pred.py --image_dir test --model model_best.pth.tar --output_csvpath result.csv --cuda Step to create visualztions on predictions from model python Resultvisualization.py