diff --git a/README.md b/README.md index b82ab2a..43deb9d 100644 --- a/README.md +++ b/README.md @@ -18,6 +18,7 @@ Usage Requirements: - [pytorch](https://github.com/pytorch/pytorch) +- [numpy](https://github.com/numpy/numpy) - [hydra](https://github.com/facebookresearch/hydra) After cloning the repository, a GON can be trained using the `train_gon.py` script: @@ -29,7 +30,7 @@ python train_gon.py dataset.name= dataset.root=