>>> sonnet.__version__ '1.27' >>> tensorflow.__version__ '1.12.0' 

$ 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.