It seems to me, you should replace model by models in this string in this line of usage script:
model
models
generator = task.build_generator(model, cfg)
Then it will look like:
generator = task.build_generator(models, cfg)
· Sign up or log in to comment