added command line parameter selection to mnist_train_forget.py