Update readmes

README.md

@@ -127,7 +127,10 @@ For debugging in VSCode, this configuration example might be helpful to you:
 
 ## Running analyses
 
-We also provide analysis and plotting code to replicate many of the plots in our paper. See `tasks/.../analysis/*` for more details on that. We als provide some data (e.g., the mazes we generated for training) and checkpoints (see [here](#checkpoints-and-data))
+We also provide analysis and plotting code to replicate many of the plots in our paper. See `tasks/.../analysis/*` for more details on that. We als0 provide some data (e.g., the mazes we generated for training) and checkpoints (see [here](#checkpoints-and-data)). Note that ffmpeg is required for generating mp4 files from the analysis scripts. It can be installed with:
+```
+conda install -c conda-forge ffmpeg
+```
 
 
 ## Checkpoints and data

tasks/image_classification/analysis/README.md

@@ -1,12 +1,7 @@
 # Analysis
 
-This folder contains analysis code for image classifcation experiments. To build GIFs for imagenet run (from the base directory):
+This folder contains the analysis code for the image classifcation experiments. Running the following from the base directory will generate figures, gifs and mp4 files:
 
 ```
-python -m tasks.image_classification.analysis.build_imagenet_viz
-```
-
-To build the plots in the paper run:
-```
-python -m tasks.image_classification.analysis.imagenet_evaluate_and_plot
+python -m tasks.image_classification.analysis.run_imagenet_analysis
 ```

tasks/rl/README.md

@@ -7,6 +7,11 @@ To run the RL training that we used for the paper, run bash scripts from the roo
 bash tasks/rl/scripts/acrobot/train_ctm_2.sh
 ```
 
+Note that tensorboard is used for monitoring training. It should be installed with:
+```
+pip install tensorboard
+```
+
 
 ## Analysis
 To run the analysis, first make sure the checkpoints are saved in the log directory (specified by the `log_dir` argument). The checkpoints can be obtained by either running the training code, or downloading them from [this link](https://drive.google.com/file/d/1VRl6qA5lX690A1X0emNg0nRH758XJEXJ/view?usp=drive_link).