Hugging Face's logo

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

App Files Community
Ubuntu commited on
Commit
baf82b7
·
1 Parent(s): babc59e

init

Browse files
Files changed (3) hide show
  1. .ipynb_checkpoints/app-checkpoint.py +2 -2
  2. app.py +2 -2
  3. lightning_logs/placeholder.txt +0 -0
.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=8, collate_fn=collator, num_workers=8, shuffle=True)
87
- test_loader = DataLoader(test_ds, batch_size=8, 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=8, collate_fn=collator, num_workers=0, shuffle=True)
87
+ test_loader = DataLoader(test_ds, batch_size=8, collate_fn=collator, num_workers=0)
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=8, collate_fn=collator, num_workers=8, shuffle=True)
87
- test_loader = DataLoader(test_ds, batch_size=8, 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=8, collate_fn=collator, num_workers=0, shuffle=True)
87
+ test_loader = DataLoader(test_ds, batch_size=8, collate_fn=collator, num_workers=0)
88
 
89
 
90
  for name, param in model.named_parameters():
lightning_logs/placeholder.txt ADDED
File without changes