diff --git a/torch_training/bla.py b/torch_training/bla.py index aaa3c0af416352ffc4c609a022f50477c2b2ebc8..a61f7073c6759708ba405a7803ca5f3b6c983ce1 100644 --- a/torch_training/bla.py +++ b/torch_training/bla.py @@ -18,4 +18,22 @@ from torch_training.dueling_double_dqn import Agent from utils.observation_utils import norm_obs_clip, split_tree print("multi_agent_trainging.py (1)") -print("bla") \ No newline at end of file + +def main(argv): + try: + opts, args = getopt.getopt(argv, "n:", ["n_trials="]) + except getopt.GetoptError: + print('training_navigation.py -n <n_trials>') + sys.exit(2) + for opt, arg in opts: + if opt in ('-n', '--n_trials'): + n_trials = int(arg) + print("main1") + +print("multi_agent_trainging.py (2)") + +if __name__ == '__main__': + print("main") + main(sys.argv[1:]) + +print("multi_agent_trainging.py (3)") \ No newline at end of file