Spaces:
Running
Running
app.py
CHANGED
@@ -76,8 +76,8 @@ def video_identity(video,user_name,class_name,trainortest,ready):
|
|
76 |
|
77 |
|
78 |
|
79 |
-
train_loader = DataLoader(train_ds, batch_size=1, collate_fn=collator, num_workers=
|
80 |
-
test_loader = DataLoader(test_ds, batch_size=1, collate_fn=collator, num_workers=
|
81 |
|
82 |
|
83 |
for name, param in model.named_parameters():
|
|
|
76 |
|
77 |
|
78 |
|
79 |
+
train_loader = DataLoader(train_ds, batch_size=1, collate_fn=collator, num_workers=0, shuffle=True)
|
80 |
+
test_loader = DataLoader(test_ds, batch_size=1, collate_fn=collator, num_workers=0)
|
81 |
|
82 |
|
83 |
for name, param in model.named_parameters():
|