fix unittest

Author: LayneH

tests/test_models/test_heads/test_isa_head.py

@@ -1,7 +1,7 @@
 # Copyright (c) OpenMMLab. All rights reserved.
 import torch
 
-from mmseg.models.decode_heads import FCNHead, ISAHead
+from mmseg.models.decode_heads import ISAHead
 from .utils import to_cuda
 
 
@@ -14,10 +14,7 @@ def test_isa_head():
         num_classes=19,
         isa_channels=16,
         down_factor=(8, 8))
-    fcn_head = FCNHead(in_channels=32, channels=16, num_classes=19)
     if torch.cuda.is_available():
-        head, inputs = to_cuda(isa_head, inputs)
-        head, inputs = to_cuda(fcn_head, inputs)
-    prev_output = fcn_head(inputs)
-    output = isa_head(inputs, prev_output)
+        isa_head, inputs = to_cuda(isa_head, inputs)
+    output = isa_head(inputs)
     assert output.shape == (1, isa_head.num_classes, 45, 45)