chirmy commited on
Commit
b1ec5f8
·
verified ·
1 Parent(s): e15a52a

Update scripy_B3.py

Browse files
Files changed (1) hide show
  1. scripy_B3.py +3 -3
scripy_B3.py CHANGED
@@ -66,9 +66,9 @@ class PytorchWorker:
66
  preds = preds.cpu() # Move tensor to CPU
67
 
68
  # post process
69
- max_value = torch.max(outputs)
70
- if max_value < -20:
71
- preds[0]=1604
72
 
73
  print("preds: ", preds)
74
 
 
66
  preds = preds.cpu() # Move tensor to CPU
67
 
68
  # post process
69
+ # max_value = torch.max(outputs)
70
+ # if max_value < -20:
71
+ # preds[0]=1604
72
 
73
  print("preds: ", preds)
74