Spaces:

PlayfulTechnology
/

QARAC

Build error

PeteBleackley commited on Apr 5, 2024

Commit

e973c2b

1 Parent(s): 7758fd9

Diagnostics

Files changed (1) hide show

scripts.py CHANGED Viewed

@@ -125,6 +125,7 @@ def train_models(path,progress=gradio.Progress(track_tqdm=True)):
                                                                tokenizer)
     trainer.to(device)
     loss_fn = CombinedLoss()
     loss_fn.cuda()
     optimizer = torch.optim.NAdam(trainer.parameters(),lr=5.0e-5)

                                                                tokenizer)
     trainer.to(device)
+    trainer.devices()
     loss_fn = CombinedLoss()
     loss_fn.cuda()
     optimizer = torch.optim.NAdam(trainer.parameters(),lr=5.0e-5)