Wait, so are you getting the RuntimeError defined in your code?
Can you post a zip of your problem, code and model? I want to try running this and see if I get an error.
Can you post a zip of your problem, code and model? I want to try running this and see if I get an error.