!65 add device_num for model_lineage.

Merge pull request !65 from luopengting/lineage_device_num

mindinsight/lineagemgr/querier/query_model.py

@@ -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)

tests/st/func/lineagemgr/api/test_model_api.py

@@ -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': {

tests/ut/lineagemgr/querier/test_querier.py

@@ -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,