Hugging Face's logo

Spaces:
RashiAgarwal
/
YOLO_v3_PyTorchLightning
Paused

App Files Community
RashiAgarwal commited on
Commit
71b5c6c
•
1 Parent(s): f397a01

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +1 -1
config.py CHANGED
@@ -34,7 +34,7 @@ ANCHORS = [
34
 
35
  SCALED_ANCHORS = (
36
  torch.tensor(ANCHORS) * torch.tensor(S).unsqueeze(1).unsqueeze(1).repeat(1, 3, 2)
37
- ).to(device="cuda")
38
  means = [0.485, 0.456, 0.406]
39
 
40
  scale = 1.1
 
34
 
35
  SCALED_ANCHORS = (
36
  torch.tensor(ANCHORS) * torch.tensor(S).unsqueeze(1).unsqueeze(1).repeat(1, 3, 2)
37
+ )
38
  means = [0.485, 0.456, 0.406]
39
 
40
  scale = 1.1