Increase training performance

train.py

@@ -62,7 +62,7 @@ parser.add_argument('--log-interval', type=int, default=50, metavar='N',
                     help='how many batches to wait before logging training status')
 parser.add_argument('--recovery-interval', type=int, default=1000, metavar='N',
                     help='how many batches to wait before writing recovery checkpoint')
-parser.add_argument('-j', '--workers', type=int, default=2, metavar='N',
+parser.add_argument('-j', '--workers', type=int, default=6, metavar='N',
                     help='how many training processes to use (default: 1)')
 parser.add_argument('--num-gpu', type=int, default=1,
                     help='Number of GPUS to use')
@@ -113,7 +113,6 @@ def main():
     loader_train = data.DataLoader(
         dataset_train,
         batch_size=batch_size,
-        pin_memory=True,
         shuffle=True,
         num_workers=args.workers
     )
@@ -125,7 +124,6 @@ def main():
     loader_eval = data.DataLoader(
         dataset_eval,
         batch_size=4 * args.batch_size,
-        pin_memory=True,
         shuffle=False,
         num_workers=args.workers
     )