update polish and compile speed

Author: um1

model.py

@@ -27,7 +27,7 @@ def weights_init_classifier(m):
 
 class USAM(nn.Module):
     #Joint Representation Learning and Keypoint Detection for Cross-view Geo-localization. TIP2022
-    def __init__(self, kernel_size=3, padding=1, polish=False):
+    def __init__(self, kernel_size=3, padding=1, polish=True):
         super(USAM, self).__init__()
 
         kernel = torch.ones((kernel_size, kernel_size))

train.py

@@ -535,7 +535,6 @@ def draw_curve(current_epoch):
 
 if torch.cuda.get_device_capability()[0]>6 and len(opt.gpu_ids)==1 and int(version[0])>1: # should be >=7 and one gpu
     torch.set_float32_matmul_precision('high')
-    torch._dynamo.config.automatic_dynamic_shapes = True
     print("Compiling model... The first epoch may be slow, which is expected!")
     # https://huggingface.co/docs/diffusers/main/en/optimization/torch2.0
     model = torch.compile(model, mode="reduce-overhead", dynamic = True) # pytorch 2.0