CNN model expects dropout_input need to add the argument to the predict function parser.add_argument('--dropout_input', type=float, default=0.1)