Spaces:
Runtime error
Runtime error
Ubuntu
commited on
Commit
·
48fcc41
1
Parent(s):
3e91da1
fixes
Browse files- .ipynb_checkpoints/app-checkpoint.py +2 -2
- app.py +2 -2
.ipynb_checkpoints/app-checkpoint.py
CHANGED
|
@@ -82,8 +82,8 @@ def video_identity(video,user_name,class_name,trainortest,ready):
|
|
| 82 |
|
| 83 |
|
| 84 |
|
| 85 |
-
train_loader = DataLoader(train_ds, batch_size=
|
| 86 |
-
test_loader = DataLoader(test_ds, batch_size=
|
| 87 |
|
| 88 |
val_batch = next(iter(test_loader))
|
| 89 |
outputs = model(**val_batch)
|
|
|
|
| 82 |
|
| 83 |
|
| 84 |
|
| 85 |
+
train_loader = DataLoader(train_ds, batch_size=8, collate_fn=collator, num_workers=8, shuffle=True)
|
| 86 |
+
test_loader = DataLoader(test_ds, batch_size=8, collate_fn=collator, num_workers=8)
|
| 87 |
|
| 88 |
val_batch = next(iter(test_loader))
|
| 89 |
outputs = model(**val_batch)
|
app.py
CHANGED
|
@@ -82,8 +82,8 @@ def video_identity(video,user_name,class_name,trainortest,ready):
|
|
| 82 |
|
| 83 |
|
| 84 |
|
| 85 |
-
train_loader = DataLoader(train_ds, batch_size=
|
| 86 |
-
test_loader = DataLoader(test_ds, batch_size=
|
| 87 |
|
| 88 |
val_batch = next(iter(test_loader))
|
| 89 |
outputs = model(**val_batch)
|
|
|
|
| 82 |
|
| 83 |
|
| 84 |
|
| 85 |
+
train_loader = DataLoader(train_ds, batch_size=8, collate_fn=collator, num_workers=8, shuffle=True)
|
| 86 |
+
test_loader = DataLoader(test_ds, batch_size=8, collate_fn=collator, num_workers=8)
|
| 87 |
|
| 88 |
val_batch = next(iter(test_loader))
|
| 89 |
outputs = model(**val_batch)
|