def run_sql_experiment(main,
mode,
include_folders=None,
log_dir=None,
exp_prefix="experiment",
exp_name=None,
**kwargs):
if exp_name is None:
exp_name = timestamp()
if log_dir is None:
log_dir = os.path.join(DEFAULT_LOG_DIR, "local",
exp_prefix.replace("_", "-"), exp_name)
if include_folders is None:
include_folders = list()
if mode == 'ec2':
include_folders.append('softqlearning')
all_symlinks = list()
for folder in include_folders:
all_symlinks.append(_create_symlink(folder))
kwargs.update(added_project_directories=all_symlinks)
run_experiment(
method_call=main,
mode=mode,
exp_prefix=exp_prefix,
exp_name=exp_name,
log_dir=log_dir,
**kwargs)
def main():
full_experiment_name = 'sawyer'
full_experiment_name += '-' + timestamp()
args = parse_args()
saved_file = args.file
run_sql_experiment(run_experiment,
mode='local',
variant=dict(file=saved_file),
exp_prefix='sawyer' + '/' + 'reuse' + '/' + timestamp(),
exp_name=full_experiment_name,
n_parallel=1,
seed=1,
terminate_machine=True,
log_dir=None,
snapshot_mode='gap',
snapshot_gap=100,
sync_s3_pkl=True)
return
def parse_args():
parser = argparse.ArgumentParser()
parser.add_argument(
'--env', type=str, choices=AVAILABLE_ENVS, default=DEFAULT_ENV)
parser.add_argument('--exp_name', type=str, default=timestamp())
parser.add_argument('--mode', type=str, default='local')
parser.add_argument('--log_dir', type=str, default=None)
args = parser.parse_args()
return args
def main():
full_experiment_name = 'swimmer'
full_experiment_name += '-' + timestamp()
args = parse_args()
saved_file = args.file
run_sql_experiment(
run_experiment,
mode='local',
variant=dict(file=saved_file),
exp_prefix='swimmer' + '/' + 'reuse' + '/' + timestamp(),
exp_name=full_experiment_name,
n_parallel=1,
seed=1,
terminate_machine=True,
log_dir=None,
snapshot_mode='gap',
snapshot_gap=100,
sync_s3_pkl=True)
return
def parse_args():
parser = argparse.ArgumentParser()
parser.add_argument(
'--env', type=str, choices=AVAILABLE_ENVS, default=DEFAULT_ENV)
parser.add_argument('--exp_name', type=str, default=timestamp())
parser.add_argument('--mode', type=str, default='local')
parser.add_argument('--log_dir', type=str, default=None)
args = parser.parse_args()
return args