main_shell2.py

#!/usr/bin/env python
"""
run the main2.py file for multiple times (runs)

Author: Qiong
"""

# import os
import pickle

# import main as main
import sys

N_RUN = 3
i_run = 0

while i_run < N_RUN:
    # execute the code for N_RUN times

    print("********Run {} starts********".format(i_run))

    if i_run == 0:
        sys.argv = ['main.py', '--seed=1']
    elif i_run == 1:
        sys.argv = ['main.py', '--seed=21']
    elif i_run == 2:
        sys.argv = ['main.py', '--seed=42']

    exec(open("main2.py").read())
    # execfile('main2.py')  # removed in python 3.x

    with open("saved/prio_memo_e002_May_train_time_run{}.data".format(i_run), "wb") as fp:
        pickle.dump(prio_memory, fp)
    # save reward to json file
    with open("saved/prio_reward_e002_May_train_time_run{}.data".format(i_run), "wb") as fp:
        pickle.dump(prio_reward, fp)

    i_run += 1