From 749ff00a754c3fb40f710dbbb23c92833ca005b7 Mon Sep 17 00:00:00 2001 From: Stanislaw Adaszewski Date: Thu, 16 Jul 2020 20:47:57 +0200 Subject: [PATCH] Shuffle. --- experiments/decagon_run/decagon_run.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/experiments/decagon_run/decagon_run.py b/experiments/decagon_run/decagon_run.py index 911caaa..74e1a03 100644 --- a/experiments/decagon_run/decagon_run.py +++ b/experiments/decagon_run/decagon_run.py @@ -118,7 +118,7 @@ def main(): # model = torch.nn.DataParallel(model, ['cuda:0', 'cuda:1']) _wrap(TrainLoop, 'build') _wrap(TrainLoop, 'run_epoch') - loop = TrainLoop(model, batch_size=1000000) + loop = TrainLoop(model, batch_size=512, shuffle=True) loop.run_epoch()