Update src/facerender/animate.py

src/facerender/animate.py

@@ -112,25 +112,25 @@ class AnimateFromCoeff():
             he_estimator.load_state_dict(he_state_dict)
         if discriminator is not None:
             try:
-               discriminator =adjust_state_dict(checkpoint['discriminator'],discriminator)
-               discriminator.load_state_dict(discriminator)
+               discriminator_dict =adjust_state_dict(checkpoint['discriminator'],discriminator)
+               discriminator.load_state_dict(discriminator_dict)
             except:
                print ('No discriminator in the state-dict. Dicriminator will be randomly initialized')
         if optimizer_generator is not None:
-            optimizer_generator =adjust_state_dict(checkpoint['optimizer_generator'],optimizer_generator)
-            optimizer_generator.load_state_dict(optimizer_generator)
+            optimizer_generator_dict =adjust_state_dict(checkpoint['optimizer_generator'],optimizer_generator)
+            optimizer_generator.load_state_dict(optimizer_generator_dict)
         if optimizer_discriminator is not None:
             try:
-                optimizer_discriminator = adjust_state_dict(checkpoint['optimizer_discriminator'],optimizer_discriminator)
-                optimizer_discriminator.load_state_dict(optimizer_discriminator)
+                optimizer_discriminator_dict = adjust_state_dict(checkpoint['optimizer_discriminator'],optimizer_discriminator)
+                optimizer_discriminator.load_state_dict(optimizer_discriminator_dict)
             except RuntimeError as e:
                 print ('No discriminator optimizer in the state-dict. Optimizer will be not initialized')
         if optimizer_kp_detector is not None:
-            optimizer_kp_detector = adjust_state_dict(checkpoint['optimizer_kp_detector'],optimizer_kp_detector)
-            optimizer_kp_detector.load_state_dict(optimizer_kp_detector)
+            optimizer_kp_detector_dict = adjust_state_dict(checkpoint['optimizer_kp_detector'],optimizer_kp_detector)
+            optimizer_kp_detector.load_state_dict(optimizer_kp_detector_dict)
         if optimizer_he_estimator is not None:
-            optimizer_he_estimator = adjust_state_dict(checkpoint['optimizer_he_estimator'],optimizer_he_estimator)
-            optimizer_he_estimator.load_state_dict(optimizer_he_estimator)
+            optimizer_he_estimator_dict = adjust_state_dict(checkpoint['optimizer_he_estimator'],optimizer_he_estimator)
+            optimizer_he_estimator.load_state_dict(optimizer_he_estimator_dict)
 
         return checkpoint['epoch']
     
@@ -149,17 +149,17 @@ class AnimateFromCoeff():
             return new_state_dict
         
         if mapping is not None:
-            mapping = adjust_state_dict(checkpoint['mapping'],mapping)
-            mapping.load_state_dict(mapping)
+            mapping_dict = adjust_state_dict(checkpoint['mapping'],mapping)
+            mapping.load_state_dict(mapping_dict)
         if discriminator is not None:
-            discriminator = adjust_state_dict(checkpoint['discriminator'],discriminator)
-            discriminator.load_state_dict(discriminator)
+            discriminator_dict = adjust_state_dict(checkpoint['discriminator'],discriminator)
+            discriminator.load_state_dict(discriminator_dict)
         if optimizer_mapping is not None:
-            optimizer_mapping = adjust_state_dict(checkpoint['optimizer_mapping'],optimizer_mapping)
-            optimizer_mapping.load_state_dict(optimizer_mapping)
+            optimizer_mapping_dict = adjust_state_dict(checkpoint['optimizer_mapping'],optimizer_mapping)
+            optimizer_mapping.load_state_dict(optimizer_mapping_dict)
         if optimizer_discriminator is not None:
-            optimizer_discriminator= adjust_state_dict(checkpoint['optimizer_discriminator'],optimizer_discriminator)
-            optimizer_discriminator.load_state_dict(optimizer_discriminator)
+            optimizer_discriminator_dict = adjust_state_dict(checkpoint['optimizer_discriminator'],optimizer_discriminator)
+            optimizer_discriminator.load_state_dict(optimizer_discriminator_dict)
 
         return checkpoint['epoch']