Merge branch 'master' into learning-rate-schedule

.dockerignore

@@ -6,6 +6,7 @@
 __pycache__/
 
 # project
+saved/
 */saved/
 dataset/
 config/

dlk/backends/src/tb/unit_test.cpp

@@ -25,7 +25,7 @@ bool parse_input_type(int argc, char const *argv[], input_type &in_type) {
 
   if (argc != 2) {
     cout << "Error: 1st arg " << argv[0] << " is not supported..." << endl;
-    cout << "Available input type: <sequential|random|add_1>" << endl;
+    cout << "Available input type: <sequential|random|all_1>" << endl;
     flag = false;
   }
 
@@ -36,7 +36,7 @@ bool parse_input_type(int argc, char const *argv[], input_type &in_type) {
   } else if (std::strcmp(argv[1], "all_1") == 0) {
     in_type = ALL_1;
   } else {
-    cout << "Error: input type is not supported." << endl << "Available input type: <sequential|random|add_1>" << endl;
+    cout << "Error: input type is not supported." << endl << "Available input type: <sequential|random|all_1>" << endl;
     flag = false;
   }
 

lmnet/lmnet/datasets/image_folder.py

@@ -60,6 +60,7 @@ def classes(self):
         """Returns the classes list in the data set."""
 
         classes = os.listdir(self.data_dir)
+        classes = [class_name for class_name in classes if class_name != ".DS_Store"]
         classes.sort(key=lambda item: item.lower())
 
         return classes