Merge branch 'master' of https://gitee.com/deipss/BERT4Rec-VAE-Pytorc…

…h-master

PaintOut.py

@@ -40,13 +40,15 @@ def data_load_bert_cnn():
     for root, dirs, files in os.walk(dir):
         for sub_dir in dirs:
             path = root + '/' + sub_dir
-            if 1 ==1  and '25_' in sub_dir and '20m' in sub_dir:
+            if 1 ==1  and '25_' in sub_dir:
                 try:
                     data, data_config = {}, {}
                     data_file = open(path + '/logs/' + 'test_metrics.json')
                     data = json.load(data_file)
                     data_file = open(path + '/' + 'config.json')
                     data_config = json.load(data_file)
+                    if not data_config['dim'] in [8,16]:
+                        continue
 
                     print('%12s\t%3s\t%10s\t%d\t%d\t%s'
                           % (data_config['model_code'], data_config['dim'], data_config['dataset_code'],