Hugging Face's logo

Spaces:
dillonlaird
/
auto-labeler
Sleeping

App Files Community
dillonlaird commited on
Commit
80c2e04
•
1 Parent(s): d9697ef

changed hyperparameters for mobile sam

Browse files
Files changed (1) hide show
  1. app/model.py +1 -1
app/model.py CHANGED
@@ -44,7 +44,7 @@ def get_multi_label_predictor(
44
  start = time.perf_counter()
45
  weights, target_feat = train(sam, [image], [mask])
46
  print(f"training time {time.perf_counter() - start}")
47
- per_sam_model = PerSAM(sam, target_feat, 100, 0.6, 0.2, weights)
48
  return per_sam_model
49
 
50
 
 
44
  start = time.perf_counter()
45
  weights, target_feat = train(sam, [image], [mask])
46
  print(f"training time {time.perf_counter() - start}")
47
+ per_sam_model = PerSAM(sam, target_feat, 10, 0.4, 0.2, weights)
48
  return per_sam_model
49
 
50