Add sqrt and scale to amp black list for msvsr_reds (#736)

* Add sqrt and scale to amp black list for msvsr_reds

* Update requirements.txt

ppgan/models/msvsr_model.py

@@ -118,7 +118,7 @@ class MultiStageVSRModel(BaseSRModel):
                     net.find_unused_parameters = False
 
         # put loss computation in amp context
-        with paddle.amp.auto_cast(enable=True, level=amp_level):
+        with paddle.amp.auto_cast(enable=True, custom_black_list={'sqrt','scale'}, level=amp_level):
             output = self.nets['generator'](self.lq)
             if isinstance(output, (list, tuple)):
                 out_stage2, output = output

requirements.txt

@@ -2,7 +2,7 @@ tqdm
 PyYAML>=5.1
 scikit-image>=0.14.0
 scipy>=1.1.0
-opencv-python
+opencv-python <= 4.6.0
 imageio==2.9.0
 imageio-ffmpeg
 librosa==0.8.1