sgoodfriend's picture
A2C playing BreakoutNoFrameskip-v4 from https://github.com/sgoodfriend/rl-algo-impls/tree/983cb75e43e51cf4ef57f177194ab9a4a1a8808b
233c511
# Support for PyTorch mps mode (https://pytorch.org/docs/stable/notes/mps.html)
import os
os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
from rl_algo_impls.runner.running_utils import base_parser
from rl_algo_impls.runner.selfplay_evaluate import SelfplayEvalArgs, selfplay_evaluate
def selfplay_enjoy() -> None:
parser = base_parser(multiple=False)
parser.add_argument(
"--wandb-run-paths",
type=str,
nargs="*",
help="WandB run paths to load players from. Must be 0 or 2",
)
parser.add_argument(
"--model-file-paths",
type=str,
help="File paths to load players from. Must be 0 or 2",
)
parser.add_argument("--render", action="store_true")
parser.add_argument("--n-envs", default=1, type=int)
parser.add_argument("--n-episodes", default=1, type=int)
parser.add_argument("--deterministic-eval", default=None, type=bool)
parser.add_argument(
"--no-print-returns", action="store_true", help="Limit printing"
)
parser.add_argument(
"--video-path", type=str, help="Path to save video of all plays"
)
# parser.set_defaults(
# algo=["ppo"],
# env=["Microrts-selfplay-unet-decay"],
# n_episodes=10,
# model_file_paths=[
# "downloaded_models/ppo-Microrts-selfplay-unet-decay-S3-best",
# "downloaded_models/ppo-Microrts-selfplay-unet-decay-S2-best",
# ],
# video_path="/Users/sgoodfriend/Desktop/decay3-vs-decay2",
# )
args = parser.parse_args()
args.algo = args.algo[0]
args.env = args.env[0]
args.seed = args.seed[0]
args = SelfplayEvalArgs(**vars(args))
selfplay_evaluate(args, os.getcwd())
if __name__ == "__main__":
selfplay_enjoy()