Skip to content

Commit 67e33ea

Browse files
committed
Reduce SAM test precision
1 parent fa2657f commit 67e33ea

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/models/sam/test_modeling_sam.js

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ export default () => {
2828
const { pred_masks, iou_scores } = await model(inputs);
2929

3030
expect(pred_masks.dims).toEqual([1, 1, 3, 256, 256]);
31-
expect(pred_masks.mean().item()).toBeCloseTo(-5.769824981689453, 5);
31+
expect(pred_masks.mean().item()).toBeCloseTo(-5.769824981689453, 3);
3232
expect(iou_scores.dims).toEqual([1, 1, 3]);
3333
expect(iou_scores.tolist()).toBeCloseToNested([[[0.8583833575248718, 0.9773167967796326, 0.8511142730712891]]]);
3434

0 commit comments

Comments
 (0)