这段代码该怎么理解呀,查都查不到
models = {} def get_model(name, num_cls=10, **args): net = models[name](num_cls=num_cls, **args) if torch.cuda.is_available(): net = net.cuda() return net