Solved by ptrblck in post #6 
              
              
                The error message sounds like you already changed the conv_class layer. 
Could you check, if you are using two different layers as the output, since the error points to conv_class, while you are manipulating conv_classify. 
Here is a small code snippet of what I was thinking about: 
class MyModel(nn…