a

.ipynb_checkpoints/app-checkpoint.py

@@ -94,7 +94,7 @@ def video_identity(video,user_name,class_name,trainortest,ready):
                 
         pl.seed_everything(42)
         classifier = Classifier(model, lr=2e-5)
-        trainer = pl.Trainer(accelerator='cpu', devices=1, precision=16, max_epochs=3)
+        trainer = pl.Trainer(accelerator='gpu', devices=1, precision=16, max_epochs=3)
         
         trainer.fit(classifier, train_loader, test_loader)
         

app.py

@@ -94,7 +94,7 @@ def video_identity(video,user_name,class_name,trainortest,ready):
                 
         pl.seed_everything(42)
         classifier = Classifier(model, lr=2e-5)
-        trainer = pl.Trainer(accelerator='cpu', devices=1, precision=16, max_epochs=3)
+        trainer = pl.Trainer(accelerator='gpu', devices=1, precision=16, max_epochs=3)
         
         trainer.fit(classifier, train_loader, test_loader)