oskarastrom commited on
Commit
cf87b0c
·
1 Parent(s): c376f3c

Update InferenceConfig.py

Browse files
Files changed (1) hide show
  1. InferenceConfig.py +3 -0
InferenceConfig.py CHANGED
@@ -56,9 +56,12 @@ class InferenceConfig:
56
  self.byte_high_conf = high
57
 
58
  def find_model(self, model_list):
 
59
  for model_name, model_path in enumerate(model_list):
 
60
  if model_path == self.weights:
61
  return model_name
 
62
  return None
63
 
64
 
 
56
  self.byte_high_conf = high
57
 
58
  def find_model(self, model_list):
59
+ print("weights", self.weights)
60
  for model_name, model_path in enumerate(model_list):
61
+ print("Path", model_path, "->", model_name)
62
  if model_path == self.weights:
63
  return model_name
64
+ print("not found")
65
  return None
66
 
67