wide&deep only save 0ckpt in data parallel

model_zoo/official/recommend/wide_and_deep/requirements.txt

@@ -0,0 +1,3 @@
+numpy
+pandas
+sklearn

model_zoo/official/recommend/wide_and_deep/train_and_eval_distribute.py

@@ -109,8 +109,11 @@ def train_and_eval(config):
                                      directory=config.ckpt_path, config=ckptconfig)
     out = model.eval(ds_eval)
     print("=====" * 5 + "model.eval() initialized: {}".format(out))
+    callback_list = [TimeMonitor(ds_train.get_dataset_size()), eval_callback, callback]
+    if get_rank() == 0:
+        callback_list.append(ckpoint_cb)
     model.train(epochs, ds_train,
-                callbacks=[TimeMonitor(ds_train.get_dataset_size()), eval_callback, callback, ckpoint_cb],
+                callbacks=callback_list,
                 sink_size=ds_train.get_dataset_size())
 
 

model_zoo/official/recommend/wide_and_deep_multitable/requirements.txt

@@ -1,3 +1,4 @@
 numpy
 pandas
 pickle
+sklearn