Skip to content

Commit

Permalink
changed argument in adaptformer test
Browse files Browse the repository at this point in the history
  • Loading branch information
caroteu committed Nov 13, 2024
1 parent e98e8ed commit 6c873d2
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion test/test_models/test_peft_sam.py
Original file line number Diff line number Diff line change
Expand Up @@ -82,7 +82,7 @@ def test_adaptformer_peft_sam(self):
from micro_sam.models.peft_sam import PEFT_Sam, AdaptFormer

_, sam = util.get_sam_model(model_type=self.model_type, return_sam=True, device="cpu")
peft_sam = PEFT_Sam(sam, rank=2, peft_module=AdaptFormer, alpha='2.0')
peft_sam = PEFT_Sam(sam, rank=2, peft_module=AdaptFormer, projection_size=64, alpha=2.0, dropout=0.5)

shape = (3, 1024, 1024)
expected_shape = (1, 3, 1024, 1024)
Expand Down

0 comments on commit 6c873d2

Please sign in to comment.