decrease eval frames sample size

evaluate.py

@@ -131,7 +131,7 @@ def get_score(item, image_paths, video_path, metrics, train_steps=100, inference
                 gt_frames.append(img)
     else:
         gt_frames = extract_frames(video_path, fps)
-        gt_frames = gt_frames[:500]
+        gt_frames = gt_frames[:200]
         for f in gt_frames:
             f.thumbnail((512,512))
 
@@ -168,7 +168,7 @@ def get_score(item, image_paths, video_path, metrics, train_steps=100, inference
         psnr2.append(float(compute_psnr(gt, base)))
         lpips2.append(float(compute_lpips(gt, base)))
         
-        if c<100:
+        if c<50:
             print(c)
             fid.append(float(compute_fid(gt, result)))
             fid2.append(float(compute_fid(gt, base)))

main.py

@@ -85,7 +85,7 @@ debug = False
 save_model = True
 should_gen_vid = False
 max_batch_size = 8
-max_frame_count = 500
+max_frame_count = 200
 
 def save_temp_imgs(imgs):
     os.makedirs('temp', exist_ok=True)