Update main_test_SUNet.py

main_test_SUNet.py

@@ -61,10 +61,7 @@ def main():
             square_input_, mask, max_wh = overlapped_square(input_.to(device), kernel=model_img, stride=stride)
             output_patch = torch.zeros(square_input_[0].shape).type_as(square_input_[0])
             for i, data in enumerate(square_input_):
-                s = time.time()
                 restored = model(square_input_[i])
-                f = time.time()
-                print(f-s)
                 if i == 0:
                     output_patch += restored
                 else: