We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 99e96f9 commit a0f61e6Copy full SHA for a0f61e6
tests/models/test_image_recog_model.py
@@ -1,5 +1,5 @@
1
import torch
2
-from onepiece_classify.models.build_model import image_recog
+from onepiece_classify.models import image_recog
3
4
def test_model():
5
rand_tensor = torch.rand([1, 3, 224, 224])
0 commit comments