NameError: name 'mnist_normalize' is not defined

[maxerbubba]

Command run:

python cnn.py --network_type backprop --dataset mnist --learning_rate 0.00025 --n_inference_steps 100 --logdir  ./logs  --savedir  ./logs  --old_savedir  None

Output:

Initialized
Args parsed
folders created
Traceback (most recent call last):
  File "cnn.py", line 268, in <module>
    dataset,testset = get_cnn_dataset(args.dataset,args.batch_size)
  File "/Volumes/workplace/ml/pressing/PredictiveCodingBackprop/datasets.py", line 62, in get_cnn_dataset
    mnist_transform = transforms.Compose([transforms.ToTensor(), mnist_normalize])
NameError: name 'mnist_normalize' is not defined

Problem location:

PredictiveCodingBackprop/datasets.py

Line 62 in 2154d78

mnist_transform = transforms.Compose([transforms.ToTensor(), mnist_normalize])

Did you mean to do something like this?
https://github.com/HazyResearch/structured-nets/blob/master/pytorch/old/misc/circtest/circulant.py#L99

mnist_normalize = transforms.Normalize((0.1307, ), (0.3081, ))