asd

app.py

@@ -11,6 +11,9 @@ import pytorch_lightning as pl
 from torch.utils.data import DataLoader
 from torchmetrics import Accuracy
 
+
+
+
 def video_identity(video,user_name,class_name,trainortest,ready):
     if ready=='yes':
 
@@ -73,8 +76,8 @@ def video_identity(video,user_name,class_name,trainortest,ready):
 
             
            
-        train_loader = DataLoader(train_ds, batch_size=8, collate_fn=collator, num_workers=8, shuffle=True)
-        test_loader = DataLoader(test_ds, batch_size=8, collate_fn=collator, num_workers=8)
+        train_loader = DataLoader(train_ds, batch_size=1, collate_fn=collator, num_workers=8, shuffle=True)
+        test_loader = DataLoader(test_ds, batch_size=1, collate_fn=collator, num_workers=8)
         
         
         for name, param in model.named_parameters():