diff --git a/docs/docs/img/design.png b/docs/docs/img/design.png index 6a950e6..c48f6ff 100644 Binary files a/docs/docs/img/design.png and b/docs/docs/img/design.png differ diff --git a/docs/docs/index.md b/docs/docs/index.md index 37cc3ef..7a79b1e 100644 --- a/docs/docs/index.md +++ b/docs/docs/index.md @@ -2,7 +2,7 @@ ## Motivation -Enable easy design and training of state-of-the-art reinforcement learning algorithms with multiple environments +Train and evaluate reinforcement learning agents by harnessing the power of multi-core CPU processing to achieve state-of-the-art results. Provide a sandbox for easing the development process of new algorithms through a modular design and an elegant set of APIs. ## Solution diff --git a/docs/mkdocs.yml b/docs/mkdocs.yml index 9825418..dc833ac 100644 --- a/docs/mkdocs.yml +++ b/docs/mkdocs.yml @@ -1,5 +1,5 @@ site_name: Reinforcement Learning Coach Documentation -theme: material +theme: readthedocs site_description: 'Reinforcement Learning Coach Documentation by Intel Nervana.' markdown_extensions: - mdx_math: diff --git a/parallel_actor.py b/parallel_actor.py index f96f9de..ac024d1 100644 --- a/parallel_actor.py +++ b/parallel_actor.py @@ -29,14 +29,6 @@ import shutil start_time = time.time() - -# Disables write_meta_graph argument, which freezes entire process and is mostly useless. -class FastSaver(tf.train.Saver): - def save(self, sess, save_path, global_step=None, latest_filename=None, - meta_graph_suffix="meta", write_meta_graph=True): - super(FastSaver, self).save(sess, save_path, global_step, latest_filename, - meta_graph_suffix, False) - if __name__ == "__main__": parser = argparse.ArgumentParser() parser.add_argument('--ps_hosts',