Self-Forcing / scripts /evaluate_three_block_fppf.py
Cccccz's picture
Upload Python scripts
bc29ee3 verified
Raw
History Blame Contribute Delete
632 Bytes
#!/usr/bin/env python3
"""Evaluate consecutive three-block Predictors with the FPPF protocol."""
from __future__ import annotations
import sys
from evaluate_two_block_fppf import main
DEFAULTS = {
"--architecture": "three_block",
"--sweep_dir": "outputs/three_block_consecutive_sweep",
"--output_dir": "outputs/three_block_fppf_eval",
"--reference_root": "outputs/single_block_fppf_eval",
"--schedule": "FPPF",
}
if __name__ == "__main__":
present = set(sys.argv[1:])
for option, value in DEFAULTS.items():
if option not in present:
sys.argv.extend([option, value])
main()