IF YOU WOULD LIKE TO GET AN ACCOUNT, please write an email to s dot adaszewski at gmail dot com. User accounts are meant only to report issues and/or generate pull requests. This is a purpose-specific Git hosting for ADARED projects. Thank you for your understanding!
sadaszewski1
/
decagon-pytorch
forked from sadaszewski/decagon-pytorch
1
0
Fork 0
Code Issues 0 Pull Requests 0 Releases 0 Wiki Activity
Browse Source

Shuffle.

master
Stanislaw Adaszewski 3 years ago
parent
b9e7e395cb
commit
749ff00a75
1 changed files with 1 additions and 1 deletions
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +1
    -1
      experiments/decagon_run/decagon_run.py

+ 1
- 1
experiments/decagon_run/decagon_run.py View File

@@ -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()


Write Preview
Loading…
Cancel
Save