提交 c1072979 编写于 作者: L luopengting

add device_num

上级 41034372
......@@ -35,6 +35,7 @@ FIELD_MAPPING = {
"learning_rate": Field('hyper_parameters', 'learning_rate'),
"epoch": Field('hyper_parameters', 'epoch'),
"batch_size": Field('hyper_parameters', 'batch_size'),
"device_num": Field('hyper_parameters', 'device_num'),
"loss": Field('algorithm', 'loss'),
"model_size": Field('model', 'size'),
"dataset_mark": Field('dataset_mark', None),
......
......@@ -76,6 +76,7 @@ LINEAGE_FILTRATION_EXCEPT_RUN = {
'learning_rate': 0.11999999731779099,
'epoch': 10,
'batch_size': 32,
'device_num': 2,
'loss': 0.029999999329447746,
'model_size': 64,
'metric': {},
......@@ -97,6 +98,7 @@ LINEAGE_FILTRATION_RUN1 = {
'learning_rate': 0.11999999731779099,
'epoch': 14,
'batch_size': 32,
'device_num': 2,
'loss': None,
'model_size': 64,
'metric': {
......@@ -120,6 +122,7 @@ LINEAGE_FILTRATION_RUN2 = {
'learning_rate': None,
'epoch': None,
'batch_size': None,
'device_num': None,
'loss': None,
'model_size': None,
'metric': {
......
......@@ -93,6 +93,7 @@ def create_filtration_result(summary_dir, train_event_dict,
"learning_rate": train_event_dict['train_lineage']['hyper_parameters']['learning_rate'],
"epoch": train_event_dict['train_lineage']['hyper_parameters']['epoch'],
"batch_size": train_event_dict['train_lineage']['hyper_parameters']['batch_size'],
"device_num": train_event_dict['train_lineage']['hyper_parameters']['device_num'],
"loss": train_event_dict['train_lineage']['algorithm']['loss'],
"model_size": train_event_dict['train_lineage']['model']['size'],
"metric": metric_dict,
......@@ -208,6 +209,7 @@ LINEAGE_FILTRATION_5 = {
event_data.EVENT_TRAIN_DICT_5['train_lineage']['hyper_parameters']['learning_rate'],
"epoch": event_data.EVENT_TRAIN_DICT_5['train_lineage']['hyper_parameters']['epoch'],
"batch_size": event_data.EVENT_TRAIN_DICT_5['train_lineage']['hyper_parameters']['batch_size'],
"device_num": event_data.EVENT_TRAIN_DICT_5['train_lineage']['hyper_parameters']['device_num'],
"loss": event_data.EVENT_TRAIN_DICT_5['train_lineage']['algorithm']['loss'],
"model_size": event_data.EVENT_TRAIN_DICT_5['train_lineage']['model']['size'],
"metric": {},
......@@ -231,6 +233,7 @@ LINEAGE_FILTRATION_6 = {
"learning_rate": None,
"epoch": None,
"batch_size": None,
"device_num": None,
"loss": None,
"model_size": None,
"metric": event_data.METRIC_5,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册