[Fix] Fix ldc_2d infer key order (#1087)

* update multi devices document

* fix index

* update doc

* fix ldc 2d inference key order

Author: HydrogenSulfate

examples/ldc/ldc_2d_Re1000_plain.py

@@ -278,7 +278,9 @@ def inference(cfg: DictConfig):
     # mapping data to cfg.INFER.output_keys
     output_dict = {
         store_key: output_dict[infer_key]
-        for store_key, infer_key in zip(cfg.MODEL.output_keys, output_dict.keys())
+        for store_key, infer_key in zip(
+            sorted(cfg.MODEL.output_keys), output_dict.keys()
+        )
     }
     U_pred = np.sqrt(output_dict["u"] ** 2 + output_dict["v"] ** 2).reshape(
         [len(x_star), len(y_star)]

examples/ldc/ldc_2d_Re3200_piratenet.py

@@ -293,7 +293,9 @@ def inference(cfg: DictConfig):
     # mapping data to cfg.INFER.output_keys
     output_dict = {
         store_key: output_dict[infer_key]
-        for store_key, infer_key in zip(cfg.MODEL.output_keys, output_dict.keys())
+        for store_key, infer_key in zip(
+            sorted(cfg.MODEL.output_keys), output_dict.keys()
+        )
     }
     U_pred = np.sqrt(output_dict["u"] ** 2 + output_dict["v"] ** 2).reshape(
         [len(x_star), len(y_star)]

examples/ldc/ldc_2d_Re3200_sota.py

@@ -289,7 +289,9 @@ def inference(cfg: DictConfig):
     # mapping data to cfg.INFER.output_keys
     output_dict = {
         store_key: output_dict[infer_key]
-        for store_key, infer_key in zip(cfg.MODEL.output_keys, output_dict.keys())
+        for store_key, infer_key in zip(
+            sorted(cfg.MODEL.output_keys), output_dict.keys()
+        )
     }
     U_pred = np.sqrt(output_dict["u"] ** 2 + output_dict["v"] ** 2).reshape(
         [len(x_star), len(y_star)]