Fixed error loadng model
#8
by knoriy - opened
task.build_generator(models, cfg) requires a list, this causes an error due to model = models[0]
use
generator = task.build_generator([model], cfg)
task.build_generator(models, cfg) requires a list, this causes an error due to model = models[0]
use
generator = task.build_generator([model], cfg)