Commit 60aa199b authored by jameskrw's avatar jameskrw
Browse files

minor

parent ac860cc0

scripts/exps/finegrained/frozenlake/worldmodeling/run.sh

+1 −1
Original line number Original line Diff line number Diff line
@@ -21,7 +21,7 @@ python -m vagen.env.create_dataset \ 




python3 -m vagen.trainer.main_ppo \

python3 -m vagen.trainer.main_ppo \

    algorithm.adv_estimator=bi_level_gae \

    algorithm.adv_estimator=bi_level_gae \

    algorithm.high_level_gamma=0.95 \

    algorithm.high_level_gamma=1.0 \

    data.train_files=data/$EXPERIMENT_NAME/train.parquet \

    data.train_files=data/$EXPERIMENT_NAME/train.parquet \

    data.val_files=data/$EXPERIMENT_NAME/test.parquet \

    data.val_files=data/$EXPERIMENT_NAME/test.parquet \

    data.train_batch_size=64 \

    data.train_batch_size=64 \

scripts/exps/finegrained/frozenlake/worldmodeling/run_tmux.sh

+1 −1
Original line number Original line Diff line number Diff line
@@ -72,7 +72,7 @@ tmux send-keys -t "$TRAIN_SESSION" "python -m vagen.env.create_dataset \\ 
# Then start the training

# Then start the training

tmux send-keys -t "$TRAIN_SESSION" "python3 -m vagen.trainer.main_ppo \\

tmux send-keys -t "$TRAIN_SESSION" "python3 -m vagen.trainer.main_ppo \\

    algorithm.adv_estimator=bi_level_gae \\

    algorithm.adv_estimator=bi_level_gae \\

    algorithm.high_level_gamma=0.95 \\

    algorithm.high_level_gamma=1.0 \\

    data.train_files=data/$EXPERIMENT_NAME/train.parquet \\

    data.train_files=data/$EXPERIMENT_NAME/train.parquet \\

    data.val_files=data/$EXPERIMENT_NAME/test.parquet \\

    data.val_files=data/$EXPERIMENT_NAME/test.parquet \\

    data.train_batch_size=64 \\

    data.train_batch_size=64 \\

CRA Git | Maintained and supported by SUSTech CRA and CCSE