Hugging Face's logo

Spaces:
envision-ai
/
Custom-Object-Detection
Runtime error

App Files Community
Ubuntu commited on
Commit
840eac1
•
1 Parent(s): ec17d59

a

Browse files
Files changed (2) hide show
  1. .ipynb_checkpoints/app-checkpoint.py +1 -1
  2. app.py +1 -1
.ipynb_checkpoints/app-checkpoint.py CHANGED
@@ -84,7 +84,7 @@ def video_identity(video,user_name,class_name,trainortest,ready):
84
 
85
 
86
  train_loader = DataLoader(train_ds, batch_size=2, collate_fn=collator, num_workers=8, shuffle=True)
87
- test_loader = DataLoader(test_ds, batch_size=2, collate_fn=collator, num_workers=8)
88
 
89
 
90
  for name, param in model.named_parameters():
 
84
 
85
 
86
  train_loader = DataLoader(train_ds, batch_size=2, collate_fn=collator, num_workers=8, shuffle=True)
87
+ test_loader = DataLoader(test_ds, batch_size=2, collate_fn=collator, num_workers=7)
88
 
89
 
90
  for name, param in model.named_parameters():
app.py CHANGED
@@ -84,7 +84,7 @@ def video_identity(video,user_name,class_name,trainortest,ready):
84
 
85
 
86
  train_loader = DataLoader(train_ds, batch_size=2, collate_fn=collator, num_workers=8, shuffle=True)
87
- test_loader = DataLoader(test_ds, batch_size=2, collate_fn=collator, num_workers=8)
88
 
89
 
90
  for name, param in model.named_parameters():
 
84
 
85
 
86
  train_loader = DataLoader(train_ds, batch_size=2, collate_fn=collator, num_workers=8, shuffle=True)
87
+ test_loader = DataLoader(test_ds, batch_size=2, collate_fn=collator, num_workers=7)
88
 
89
 
90
  for name, param in model.named_parameters():