$ python train.py --problem=mnist --save_path=./mnist
WARNING:tensorflow:From /root/.anaconda/lib/python3.6/site-packages/tensorflow/contrib/learn/python/learn/datasets/mnist.py:290: DataSet.__init__ (from tensorflow.contrib.learn.python.learn.datasets.mnist) is deprecated and will be removed in a future version.
Instructions for updating:
Please use alternatives such as official/mnist/dataset.py from tensorflow/models.
Optimizee variables
['mlp/linear_0/w:0', 'mlp/linear_0/b:0', 'mlp/linear_1/w:0', 'mlp/linear_1/b:0']
Problem variables
[]
Traceback (most recent call last):
File "/root/.anaconda/lib/python3.6/site-packages/tensorflow/python/util/nest.py", line 179, in assert_same_structure
_pywrap_tensorflow.AssertSameStructure(nest1, nest2, check_types)
TypeError: The two structures don't have the same nested structure.
Presumably this is a version issue -- does anyone know tensorflow/sonnet versions where this repo will run?
I'm trying to run w/ the following versions in python==3.6.5
and getting the following error:
Presumably this is a version issue -- does anyone know tensorflow/sonnet versions where this repo will run?
Thanks! ~ Ben