InvalidArgumentError after model compilation

Please see this topic. Do remove code from your post. It’s ok to leave stacktraces.