问题遇到的现象和发生背景
这段代码该怎么理解呀,查都查不到
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
这段代码该怎么理解呀,查都查不到
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