:white_check_mark: [Pass] the test with fixing model location

tests/test_model/test_yolo.py

@@ -37,8 +37,8 @@ def cfg() -> Config:
 @pytest.fixture
 def model(cfg: Config):
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-    model = create_model(cfg.model, weight_path=None)
-    return model.to(device)
+    model = create_model(cfg.model, weight_path=None, device=device)
+    return model
 
 
 def test_model_basic_status(model):

tests/test_utils/test_loss.py

@@ -24,8 +24,8 @@ def cfg() -> Config:
 @pytest.fixture
 def model(cfg: Config):
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-    model = create_model(cfg.model, weight_path=None)
-    return model.to(device)
+    model = create_model(cfg.model, weight_path=None, device=device)
+    return model
 
 
 @pytest.fixture