def exit_if_trainer_done(checkpoint_dir): '''Helper method that shutsdown the sim app if the trainer is done checkpoint_dir - direcotry where the done file would be downloaded to ''' if should_stop(checkpoint_dir): logger.info("Received termination signal from trainer. Goodbye.") utils.simapp_exit_gracefully()
def exit_if_trainer_done(checkpoint_dir, s3_writer): '''Helper method that shutsdown the sim app if the trainer is done checkpoint_dir - direcotry where the done file would be downloaded to ''' if should_stop(checkpoint_dir): unsubscribe_from_save_mp4 = ServiceProxyWrapper('/racecar/save_mp4/unsubscribe_from_save_mp4', Empty) unsubscribe_from_save_mp4(EmptyRequest()) s3_writer.upload_to_s3() logger.info("Received termination signal from trainer. Goodbye.") utils.simapp_exit_gracefully()
def exit_if_trainer_done(checkpoint_dir, s3_writer, rollout_idx): '''Helper method that shutsdown the sim app if the trainer is done checkpoint_dir - direcotry where the done file would be downloaded to ''' if should_stop(checkpoint_dir): is_save_mp4_enabled = rospy.get_param('MP4_S3_BUCKET', None) and rollout_idx == 0 if is_save_mp4_enabled: unsubscribe_from_save_mp4 = ServiceProxyWrapper('/racecar/save_mp4/unsubscribe_from_save_mp4', Empty) unsubscribe_from_save_mp4(EmptyRequest()) s3_writer.upload_to_s3() logger.info("Received termination signal from trainer. Goodbye.") simapp_exit_gracefully()
def exit_if_trainer_done(checkpoint_dir, simtrace_video_s3_writers, rollout_idx): """Helper method that shutsdown the sim app if the trainer is done checkpoint_dir - direcotry where the done file would be downloaded to """ if should_stop(checkpoint_dir): is_save_mp4_enabled = rospy.get_param("MP4_S3_BUCKET", None) and rollout_idx == 0 if is_save_mp4_enabled: unsubscribe_from_save_mp4 = ServiceProxyWrapper( "/racecar/save_mp4/unsubscribe_from_save_mp4", Empty ) unsubscribe_from_save_mp4(EmptyRequest()) # upload simtrace and mp4 into s3 bucket for s3_writer in simtrace_video_s3_writers: s3_writer.persist(utils.get_s3_kms_extra_args()) logger.info("Received termination signal from trainer. Goodbye.") simapp_exit_gracefully()