Traceback $($most recent call last$)$:

File "submission.py$",$ line $13,$ in

from train$\_$utils import batchify$\_$data, run$\_$epoch, train$\_$model, Flatten

ModuleNotFoundError: No module named 'train$\_$utils getting this error repeatedly $.$ I think this error is rising due to the part$2-$mnist and part$2-$twodigit both this subfolders are having the train$\_$utils.py$.$