From 1dce0c0d5aa2582f85188db4358ec4691a8c40ce Mon Sep 17 00:00:00 2001
From: alexlee-gk <alexleegk@gmail.com>
Date: Wed, 11 Apr 2018 11:44:43 -0700
Subject: [PATCH] Add info about hparams in the readme.

---
 README.md | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/README.md b/README.md
index 2d88db10..e5f41327 100644
--- a/README.md
+++ b/README.md
@@ -104,6 +104,11 @@ The following are ablations of our model:
 
 See [`pretrained_models/download_model.sh`](pretrained_models/download_model.sh) for a complete list of available pre-trained models.
 
+### Model and Training Hyperparameters
+The implementation is designed such that each video prediction model defines its architecture and training procedure, and include reasonable hyperparameters as defaults.
+Still, a few of the hyperparameters should be overriden for each variant of dataset and model.
+The hyperparameters used in our experiments are provided in [`hparams`](hparams) as JSON files, and they can be passed onto the training script with the `--model_hparams_dict` flag.
+
 ## Citation
 
 If you find this useful for your research, please use the following.
-- 
GitLab