Update valid_code_release.py

Author: zibojia

valid_code_release.py

@@ -98,6 +98,8 @@ def main(
 
     state_dict2 = {}
     for key in state_dict:
+        if 'pe' in key:
+            continue
         state_dict2[key.split('module.')[1]] = state_dict[key]
 
     m, u = unet.load_state_dict(state_dict2, strict=False)