Spaces:

davidserra9
/

NamedCurves

Running on Zero

davidserra9 commited on Jan 9

Commit

ea8a8a0

verified ·

1 Parent(s): 1f6d8ec

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -58,7 +58,7 @@ config = dict2namespace(config)
 device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
-config.train.cuda_visible_device = device
 model = NamedCurves(config.model).to(device)
 model.load_state_dict(torch.load(model_pt)["model_state_dict"])

 device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
+config['train']['cuda_visible_device'] = device
 model = NamedCurves(config.model).to(device)
 model.load_state_dict(torch.load(model_pt)["model_state_dict"])