#!/bin/bash

# Use 42 as default if no seed is provided
SEED=${1:-42}

python -m src.main algorithm=lstm_aurora env=ant seed=$SEED logging.tags="['baseline']" logging.group="baseline" device="cuda"
python -m src.main algorithm=lstm_aurora env=bipedal_walker seed=$SEED logging.tags="['baseline']" logging.group="baseline" device="cuda"
python -m src.main algorithm=lstm_aurora env=halfcheetah seed=$SEED logging.tags="['baseline']" logging.group="baseline" device="cuda"
python -m src.main algorithm=lstm_aurora env=hopper seed=$SEED logging.tags="['baseline']" logging.group="baseline" device="cuda"
python -m src.main algorithm=lstm_aurora env=walker2d seed=$SEED logging.tags="['baseline']" logging.group="baseline" device="cuda"
python -m src.main algorithm=lstm_aurora env=swimmer seed=$SEED logging.tags="['baseline']" logging.group="baseline" device="cuda"