diff --git a/README.md b/README.md
index e89e344baf259ae8913fe68ac10abe5eb38d2e94..931bbd71085c295e099bcb82a5d82309ae5a1be7 100644
--- a/README.md
+++ b/README.md
@@ -40,7 +40,7 @@ We use [EG3D](https://github.com/NVlabs/eg3d) as a pretrained face generator net
 ```sh
 $ git clone https://github.com/NVlabs/eg3d.git
 ```
-We use `ffhq512-128.pkl` [checkpoint](https://catalog.ngc.nvidia.com/orgs/nvidia/teams/research/models/eg3d/files) in our experiments.
+We use `ffhqrebalanced512-128.pkl` [checkpoint](https://catalog.ngc.nvidia.com/orgs/nvidia/teams/research/models/eg3d/files) in our experiments.
 
 ## Downloading the datasets
 In our experiments, we use [FFHQ](https://github.com/NVlabs/ffhq-dataset) dataset for training our face reconstruction network.
diff --git a/evaluation_pipeline.py b/evaluation_pipeline.py
index dc47eadf1d8e5aa6a4d54aee0b2a5d8d5f39b4f9..463d8afe1427f0ccc5812377422eb1ecee7274bb 100644
--- a/evaluation_pipeline.py
+++ b/evaluation_pipeline.py
@@ -12,8 +12,8 @@ parser.add_argument('--checkpoint', metavar='<checkpoint>', type= str, default='
                     help='checkpoint of the new mapping network')
 parser.add_argument('--path_eg3d_repo', metavar='<path_eg3d_repo>', type= str, default='./eg3d',
                     help='./eg3d')
-parser.add_argument('--path_eg3d_checkpoint', metavar='<path_eg3d_checkpoint>', type= str, default='./ffhq512-128.pkl',
-                    help='./ffhq512-128.pkl`')
+parser.add_argument('--path_eg3d_checkpoint', metavar='<path_eg3d_checkpoint>', type= str, default='./ffhqrebalanced512-128.pkl',
+                    help='./ffhqrebalanced512-128.pkl`')
 
 args = parser.parse_args()
 
diff --git a/train.py b/train.py
index 40bba9246c76200cf684d9043a50901b67ba32d1..f2eb6f9ef70224d0766de8057c2d62e8582a6bcf 100644
--- a/train.py
+++ b/train.py
@@ -12,8 +12,8 @@ import argparse
 parser = argparse.ArgumentParser(description='Train face reconstruction network - GaFaR')
 parser.add_argument('--path_eg3d_repo', metavar='<path_eg3d_repo>', type= str, default='./eg3d',
                     help='./eg3d')
-parser.add_argument('--path_eg3d_checkpoint', metavar='<path_eg3d_checkpoint>', type= str, default='./ffhq512-128.pkl',
-                    help='./ffhq512-128.pkl')
+parser.add_argument('--path_eg3d_checkpoint', metavar='<path_eg3d_checkpoint>', type= str, default='./ffhqrebalanced512-128.pkl',
+                    help='./ffhqrebalanced512-128.pkl')
 parser.add_argument('--path_ffhq_dataset', metavar='<path_ffhq_dataset>', type= str, default='./Flickr-Faces-HQ/images1024x1024',
                     help='FFHQ directory')
 parser.add_argument('--FR_system', metavar='<FR_system>', type= str, default='ArcFace',