From 7c53c5cbfdfd83f0e9d201215b15f59956604564 Mon Sep 17 00:00:00 2001 From: deipss Date: Tue, 25 May 2021 18:23:40 +0800 Subject: [PATCH] =?UTF-8?q?=E6=97=A5=E5=BF=97=E6=89=93=E5=8D=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- PaintOut.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/PaintOut.py b/PaintOut.py index 7aec706..1b8d86e 100644 --- a/PaintOut.py +++ b/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'],