Upload folder using huggingface_hub

Files changed (2) hide show

__pycache__/predict.cpython-311.pyc CHANGED Viewed

Binary files a/__pycache__/predict.cpython-311.pyc and b/__pycache__/predict.cpython-311.pyc differ

predict.py CHANGED Viewed

@@ -167,7 +167,7 @@ def predict_next_frame(ens, context_frames: np.ndarray) -> np.ndarray:
             predicted = torch.zeros_like(direct_pred)
             for step in range(PRED_FRAMES):
-                ar_weight = 0.9 - (step / (PRED_FRAMES - 1)) * 0.3
                 direct_weight = 1.0 - ar_weight
                 predicted[:, step] = ar_weight * ar_pred[:, step] + direct_weight * direct_pred[:, step]
@@ -227,7 +227,7 @@ def predict_next_frame(ens, context_frames: np.ndarray) -> np.ndarray:
             predicted = torch.zeros_like(direct_pred)
             for step in range(PRED_FRAMES):
-                ar_weight = 0.7 - (step / (PRED_FRAMES - 1)) * 0.4
                 direct_weight = 1.0 - ar_weight
                 predicted[:, step] = ar_weight * ar_pred[:, step] + direct_weight * direct_pred[:, step]

             predicted = torch.zeros_like(direct_pred)
             for step in range(PRED_FRAMES):
+                ar_weight = 0.8
                 direct_weight = 1.0 - ar_weight
                 predicted[:, step] = ar_weight * ar_pred[:, step] + direct_weight * direct_pred[:, step]
             predicted = torch.zeros_like(direct_pred)
             for step in range(PRED_FRAMES):
+                ar_weight = 0.5
                 direct_weight = 1.0 - ar_weight
                 predicted[:, step] = ar_weight * ar_pred[:, step] + direct_weight * direct_pred[:, step]