Update README.md

README.md

@@ -21,9 +21,17 @@ If you use this code, please cite our paper given below:
 ![](test.png)
 
 ## Prerequisites
-* TensorFlow and tqdm packages should be installed. The code is tested with Python 3.7, PyTorch 0.4.1 and Ubuntu 18.04. 
+* The code is tested with Python 3.7, PyTorch 0.4.1 and Ubuntu 18.04. 
+* Please, check out the requirements file for more information
 * For reproducibility, you can find our datasets on the following repository: . 
 
+## Hardware setup
+* The software was train and tested in a machine with the following features:
+ * Nvidia Quadro P2000
+ * 16 GB RAM
+ * 12 CPU cores
+
+
 ## Training
 The script `train.py` expects the following command line arguments:
 * `-r`, `--tf_record`: `TFRecord` file(s) for training.