exception handling in humanparam

tensorpack/callbacks/param.py

@@ -62,17 +62,25 @@ class HumanHyperParamSetter(HyperParamSetter):
         super(HumanHyperParamSetter, self).__init__(var_name)
 
     def  _get_current_value(self):
-        with open(self.file_name) as f:
-            lines = f.readlines()
-        lines = [s.strip().split(':') for s in lines]
-        dic = {str(k):float(v) for k, v in lines}
-        return dic[self.op_name]
+        try:
+            with open(self.file_name) as f:
+                lines = f.readlines()
+            lines = [s.strip().split(':') for s in lines]
+            dic = {str(k):float(v) for k, v in lines}
+            ret = dic[self.op_name]
+            return ret
+        except:
+            logger.warn(
+                "Failed to parse {} in {}".format(
+                    self.op_name, self.file_name))
+            return None
 
 class ScheduledHyperParamSetter(HyperParamSetter):
     def __init__(self, var_name, schedule):
         """
         schedule: [(epoch1, val1), (epoch2, val2), (epoch3, val3), ...]
         """
+        schedule = [(int(a), float(b)) for a, b in schedule]
         self.schedule = sorted(schedule, key=operator.itemgetter(0))
         super(ScheduledHyperParamSetter, self).__init__(var_name)
 

tensorpack/dataflow/dataset/ilsvrc.py

@@ -8,7 +8,7 @@ from ...utils.fs import mkdir_p, download
 
 __all__ = ['ILSVRCMeta']
 
-CAFFE_URL = "http://dl.caffe.berkeleyvision.org/caffe_ilsvrc12.tar.gz"
+CAFFE_ILSVRC12_URL = "http://dl.caffe.berkeleyvision.org/caffe_ilsvrc12.tar.gz"
 
 class ILSVRCMeta(object):
     def __init__(self, dir=None):
@@ -26,7 +26,7 @@ class ILSVRCMeta(object):
         return dict(enumerate(lines))
 
     def download_caffe_meta(self):
-        fpath = download(CAFFE_URL, self.dir)
+        fpath = download(CAFFE_ILSVRC12_URL, self.dir)
         tarfile.open(fpath, 'r:gz').extractall(self.dir)
 
 if __name__ == '__main__':