fix dygraph quant demo save model (#549)

demo/dygraph/quant/train.py

@@ -319,12 +319,14 @@ def compress(args):
                 paddle.save(net.state_dict(), model_prefix + ".pdparams")
                 paddle.save(opt.state_dict(), model_prefix + ".pdopt")
 
-    # load best model
-    load_dygraph_pretrain(net, os.path.join(args.model_save_dir, "best_model"))
-
     ############################################################################################################
     # 3. Save quant aware model
     ############################################################################################################
+    if paddle.distributed.get_rank() == 0:
+        # load best model
+        load_dygraph_pretrain(net,
+                              os.path.join(args.model_save_dir, "best_model"))
+
         path = os.path.join(args.model_save_dir, "inference_model", 'qat_model')
         quanter.save_quantized_model(
             net,