Skip to content

Commit

Permalink
main.py any stack testing fix, 100 trials
Browse files Browse the repository at this point in the history
  • Loading branch information
ahundt committed Aug 22, 2019
1 parent 26f0a46 commit 9b6866d
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions main.py
Original file line number Diff line number Diff line change
Expand Up @@ -529,7 +529,7 @@ def process_actions():
no_change_count = [0, 0]
trainer.clearance_log.append([trainer.iteration])
logger.write_to_log('clearance', trainer.clearance_log)
if is_testing and len(trainer.clearance_log) >= max_test_trials:
if is_testing and not place and len(trainer.clearance_log) >= max_test_trials:
exit_called = True # Exit after training thread (backprop and saving labels)

continue
Expand Down Expand Up @@ -840,7 +840,7 @@ def experience_replay(method, prev_primitive_action, prev_reward_value, trainer,

# -------------- Testing options --------------
parser.add_argument('--is_testing', dest='is_testing', action='store_true', default=False)
parser.add_argument('--max_test_trials', dest='max_test_trials', type=int, action='store', default=50, help='maximum number of test runs per case/scenario')
parser.add_argument('--max_test_trials', dest='max_test_trials', type=int, action='store', default=100, help='maximum number of test runs per case/scenario')
parser.add_argument('--test_preset_cases', dest='test_preset_cases', action='store_true', default=False)
parser.add_argument('--test_preset_file', dest='test_preset_file', action='store', default='test-10-obj-01.txt')

Expand Down

0 comments on commit 9b6866d

Please sign in to comment.