Revert "train on p percent of the data"

This reverts commit ba6c3fe.
nv-tlabs · Oct 28, 2021 · d74598c · d74598c
1 parent 1ddcf93
commit d74598c
Show file tree

Hide file tree

Showing 4 changed files with 0 additions and 130 deletions.
diff --git a/.dockerignore b/.dockerignore
diff --git a/Dockerfile b/Dockerfile
diff --git a/ngc.py b/ngc.py
diff --git a/src/train.py b/src/train.py
@@ -40,7 +40,6 @@ def train(version,
             nworkers=10,
             lr=1e-3,
             weight_decay=1e-7,
-            pdata = 1.0,
             ):
     grid_conf = {
         'xbound': xbound,
@@ -63,9 +62,6 @@ def train(version,
                                           grid_conf=grid_conf, bsz=bsz, nworkers=nworkers,
                                           parser_name='segmentationdata')
 
-    # take p% of the data
-    trainloader.dataset.ixes = trainloader.dataset.ixes[:int(pdata * len(trainloader.dataset.ixes))]
-
     device = torch.device('cpu') if gpuid < 0 else torch.device(f'cuda:{gpuid}')
 
     model = compile_model(grid_conf, data_aug_conf, outC=1)