Train your model and save it as cnn$.$th$.$ You can reuse some of the training functionality in train.py from homework $1.$ We highly recommend you incorporate the logging functionality from section $2$ into your training routine. Once you trained your model, you can optionally visualize your model's prediction using python$3-$m homework.viz$\_$prediction $[$DATASET$\_$PATH$].$