Hugging Face's logo

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

App Files Community
Ubuntu commited on
Commit
ec17d59
·
1 Parent(s): a714b9b

a

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