fixing cbas_cur functionality

mlaico · May 5, 2018 · d15f420 · d15f420
1 parent f48b8eb
commit d15f420
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/cbas.py b/cbas.py
@@ -30,8 +30,6 @@
 elif PYTHON_VERSION == 3:
     from urllib.request import urlretrieve
 
-import pycocotools.mask as maskUtils
-
 
 def _isArrayLike(obj):
     return hasattr(obj, '__iter__') and hasattr(obj, '__len__')

diff --git a/pytorch-classification/cbas_cur.py b/pytorch-classification/cbas_cur.py
@@ -36,7 +36,8 @@
 parser.add_argument('-d', '--dataset', default='cbas34', type=str)
 parser.add_argument('-j', '--workers', default=2, type=int, metavar='N',
                     help='number of data loading workers (default: 4)')
-parser.add_argument('--curr', default='none', type=str)
+parser.add_argument('--curr', default='none', type=str, metavar='CUR',
+                    help='type of curriculum')
 # Optimization options
 parser.add_argument('--epochs', default=300, type=int, metavar='N',
                     help='number of total epochs to run')