[Enhance] Make scipy as a default dependency in runtime in dev-1.x
parent
e98231ba5f
commit
925110044c
|
@ -140,8 +140,11 @@ def test_beit_init():
|
|||
}
|
||||
}
|
||||
model = BEiT(img_size=(512, 512))
|
||||
with pytest.raises(AttributeError):
|
||||
model.resize_rel_pos_embed(ckpt)
|
||||
# If scipy is installed, this AttributeError would not be raised.
|
||||
from mmengine.utils import is_installed
|
||||
if not is_installed('scipy'):
|
||||
with pytest.raises(AttributeError):
|
||||
model.resize_rel_pos_embed(ckpt)
|
||||
|
||||
# pretrained=None
|
||||
# init_cfg=123, whose type is unsupported
|
||||
|
|
|
@ -138,8 +138,11 @@ def test_mae_init():
|
|||
}
|
||||
}
|
||||
model = MAE(img_size=(512, 512))
|
||||
with pytest.raises(AttributeError):
|
||||
model.resize_rel_pos_embed(ckpt)
|
||||
# If scipy is installed, this AttributeError would not be raised.
|
||||
from mmengine.utils import is_installed
|
||||
if not is_installed('scipy'):
|
||||
with pytest.raises(AttributeError):
|
||||
model.resize_rel_pos_embed(ckpt)
|
||||
|
||||
# test resize abs pos embed
|
||||
ckpt = model.resize_abs_pos_embed(ckpt['state_dict'])
|
||||
|
|
Loading…
Reference in New Issue