cleanups

loudinthecloud · Oct 10, 2017 · e5477b1 · e5477b1
1 parent 91fb9d6
commit e5477b1
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/tasks/copytask.py b/tasks/copytask.py
@@ -1,10 +1,11 @@
 """Copy Task NTM model."""
+import random
+
 from attr import attrs, attrib, Factory
 import torch
 from torch import nn
 from torch.autograd import Variable
 from torch import optim
-import random
 import numpy as np
 
 from ntm.aio import EncapsulatedNTM

diff --git a/tasks/repeatcopytask.py b/tasks/repeatcopytask.py
@@ -1,10 +1,11 @@
 """Copy Task NTM model."""
+import random
+
 from attr import attrs, attrib, Factory
 import torch
 from torch import nn
 from torch.autograd import Variable
 from torch import optim
-import random
 import numpy as np
 
 from ntm.aio import EncapsulatedNTM

diff --git a/train.py b/train.py
@@ -208,7 +208,8 @@ def init_arguments():
     parser.add_argument('-p', '--param', action='append', default=[],
                         help='Override model params. Example: "-pbatch_size=4 -pnum_heads=2"')
     parser.add_argument('--checkpoint-interval', type=int, default=CHECKPOINT_INTERVAL,
-                        help="Checkpoint interval (default: {}). Use 0 to disable checkpointing".format(CHECKPOINT_INTERVAL))
+                        help="Checkpoint interval (default: {}). "
+                             "Use 0 to disable checkpointing".format(CHECKPOINT_INTERVAL))
     parser.add_argument('--checkpoint-path', action='store', default='./',
                         help="Path for saving checkpoint data (default: './')")
     parser.add_argument('--report-interval', type=int, default=REPORT_INTERVAL,