Add

21m400m400m/3487337.err

@@ -0,0 +1,141 @@
+0: 2023-05-10 00:12:26.642566: I tensorflow/core/platform/cpu_feature_guard.cc:193] This TensorFlow binary is optimized with oneAPI Deep Neural Network Library (oneDNN) to use the following CPU instructions in performance-critical operations:  AVX2 FMA
+0: To enable them in other operations, rebuild TensorFlow with the appropriate compiler flags.
+0: 2023-05-10 00:12:26.642574: I tensorflow/core/platform/cpu_feature_guard.cc:193] This TensorFlow binary is optimized with oneAPI Deep Neural Network Library (oneDNN) to use the following CPU instructions in performance-critical operations:  AVX2 FMA
+0: To enable them in other operations, rebuild TensorFlow with the appropriate compiler flags.
+0: 2023-05-10 00:12:26.642587: I tensorflow/core/platform/cpu_feature_guard.cc:193] This TensorFlow binary is optimized with oneAPI Deep Neural Network Library (oneDNN) to use the following CPU instructions in performance-critical operations:  AVX2 FMA
+0: To enable them in other operations, rebuild TensorFlow with the appropriate compiler flags.
+0: 2023-05-10 00:12:26.642609: I tensorflow/core/platform/cpu_feature_guard.cc:193] This TensorFlow binary is optimized with oneAPI Deep Neural Network Library (oneDNN) to use the following CPU instructions in performance-critical operations:  AVX2 FMA
+0: To enable them in other operations, rebuild TensorFlow with the appropriate compiler flags.
+0: 2023-05-10 00:12:26.642609: I tensorflow/core/platform/cpu_feature_guard.cc:193] This TensorFlow binary is optimized with oneAPI Deep Neural Network Library (oneDNN) to use the following CPU instructions in performance-critical operations:  AVX2 FMA
+0: To enable them in other operations, rebuild TensorFlow with the appropriate compiler flags.
+0: 2023-05-10 00:12:26.642625: I tensorflow/core/platform/cpu_feature_guard.cc:193] This TensorFlow binary is optimized with oneAPI Deep Neural Network Library (oneDNN) to use the following CPU instructions in performance-critical operations:  AVX2 FMA
+0: To enable them in other operations, rebuild TensorFlow with the appropriate compiler flags.
+0: 2023-05-10 00:12:26.642632: I tensorflow/core/platform/cpu_feature_guard.cc:193] This TensorFlow binary is optimized with oneAPI Deep Neural Network Library (oneDNN) to use the following CPU instructions in performance-critical operations:  AVX2 FMA
+0: To enable them in other operations, rebuild TensorFlow with the appropriate compiler flags.
+0: 2023-05-10 00:12:26.642626: I tensorflow/core/platform/cpu_feature_guard.cc:193] This TensorFlow binary is optimized with oneAPI Deep Neural Network Library (oneDNN) to use the following CPU instructions in performance-critical operations:  AVX2 FMA
+0: To enable them in other operations, rebuild TensorFlow with the appropriate compiler flags.
+0: 2023-05-10 00:12:35.492522: W tensorflow/compiler/xla/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libcudart.so.11.0'; dlerror: libcudart.so.11.0: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /pfs/lustrep2/projappl/project_462000125/samantao-public/apps-rocm-5.2.3/aws-ofi-rccl:/opt/rocm/lib64:/opt/rocm/lib:/opt/rocm/rocprofiler/lib:/opt/rocm/rocprofiler/tool:/opt/rocm/roctracer/lib:/opt/rocm/roctracer/tool:/opt/rocm/hip/lib:/opt/cray/pe/python/3.9.13.1/lib:/opt/cray/pe/gcc-libs:/opt/cray/libfabric/1.15.2.0/lib64
+0: 2023-05-10 00:12:35.492558: W tensorflow/compiler/xla/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libcudart.so.11.0'; dlerror: libcudart.so.11.0: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /pfs/lustrep2/projappl/project_462000125/samantao-public/apps-rocm-5.2.3/aws-ofi-rccl:/opt/rocm/lib64:/opt/rocm/lib:/opt/rocm/rocprofiler/lib:/opt/rocm/rocprofiler/tool:/opt/rocm/roctracer/lib:/opt/rocm/roctracer/tool:/opt/rocm/hip/lib:/opt/cray/pe/python/3.9.13.1/lib:/opt/cray/pe/gcc-libs:/opt/cray/libfabric/1.15.2.0/lib64
+0: 2023-05-10 00:12:35.492577: W tensorflow/compiler/xla/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libcudart.so.11.0'; dlerror: libcudart.so.11.0: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /pfs/lustrep2/projappl/project_462000125/samantao-public/apps-rocm-5.2.3/aws-ofi-rccl:/opt/rocm/lib64:/opt/rocm/lib:/opt/rocm/rocprofiler/lib:/opt/rocm/rocprofiler/tool:/opt/rocm/roctracer/lib:/opt/rocm/roctracer/tool:/opt/rocm/hip/lib:/opt/cray/pe/python/3.9.13.1/lib:/opt/cray/pe/gcc-libs:/opt/cray/libfabric/1.15.2.0/lib64
+0: 2023-05-10 00:12:35.492623: W tensorflow/compiler/xla/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libcudart.so.11.0'; dlerror: libcudart.so.11.0: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /pfs/lustrep2/projappl/project_462000125/samantao-public/apps-rocm-5.2.3/aws-ofi-rccl:/opt/rocm/lib64:/opt/rocm/lib:/opt/rocm/rocprofiler/lib:/opt/rocm/rocprofiler/tool:/opt/rocm/roctracer/lib:/opt/rocm/roctracer/tool:/opt/rocm/hip/lib:/opt/cray/pe/python/3.9.13.1/lib:/opt/cray/pe/gcc-libs:/opt/cray/libfabric/1.15.2.0/lib64
+0: 2023-05-10 00:12:35.492620: W tensorflow/compiler/xla/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libcudart.so.11.0'; dlerror: libcudart.so.11.0: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /pfs/lustrep2/projappl/project_462000125/samantao-public/apps-rocm-5.2.3/aws-ofi-rccl:/opt/rocm/lib64:/opt/rocm/lib:/opt/rocm/rocprofiler/lib:/opt/rocm/rocprofiler/tool:/opt/rocm/roctracer/lib:/opt/rocm/roctracer/tool:/opt/rocm/hip/lib:/opt/cray/pe/python/3.9.13.1/lib:/opt/cray/pe/gcc-libs:/opt/cray/libfabric/1.15.2.0/lib64
+0: 2023-05-10 00:12:35.492624: W tensorflow/compiler/xla/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libcudart.so.11.0'; dlerror: libcudart.so.11.0: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /pfs/lustrep2/projappl/project_462000125/samantao-public/apps-rocm-5.2.3/aws-ofi-rccl:/opt/rocm/lib64:/opt/rocm/lib:/opt/rocm/rocprofiler/lib:/opt/rocm/rocprofiler/tool:/opt/rocm/roctracer/lib:/opt/rocm/roctracer/tool:/opt/rocm/hip/lib:/opt/cray/pe/python/3.9.13.1/lib:/opt/cray/pe/gcc-libs:/opt/cray/libfabric/1.15.2.0/lib64
+0: 2023-05-10 00:12:35.492635: W tensorflow/compiler/xla/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libcudart.so.11.0'; dlerror: libcudart.so.11.0: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /pfs/lustrep2/projappl/project_462000125/samantao-public/apps-rocm-5.2.3/aws-ofi-rccl:/opt/rocm/lib64:/opt/rocm/lib:/opt/rocm/rocprofiler/lib:/opt/rocm/rocprofiler/tool:/opt/rocm/roctracer/lib:/opt/rocm/roctracer/tool:/opt/rocm/hip/lib:/opt/cray/pe/python/3.9.13.1/lib:/opt/cray/pe/gcc-libs:/opt/cray/libfabric/1.15.2.0/lib64
+0: 2023-05-10 00:12:35.492657: W tensorflow/compiler/xla/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libcudart.so.11.0'; dlerror: libcudart.so.11.0: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /pfs/lustrep2/projappl/project_462000125/samantao-public/apps-rocm-5.2.3/aws-ofi-rccl:/opt/rocm/lib64:/opt/rocm/lib:/opt/rocm/rocprofiler/lib:/opt/rocm/rocprofiler/tool:/opt/rocm/roctracer/lib:/opt/rocm/roctracer/tool:/opt/rocm/hip/lib:/opt/cray/pe/python/3.9.13.1/lib:/opt/cray/pe/gcc-libs:/opt/cray/libfabric/1.15.2.0/lib64
+0: 2023-05-10 00:12:35.493431: I tensorflow/compiler/xla/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine.
+0: 2023-05-10 00:12:35.493451: I tensorflow/compiler/xla/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine.
+0: 2023-05-10 00:12:35.493467: I tensorflow/compiler/xla/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine.
+0: 2023-05-10 00:12:35.493463: I tensorflow/compiler/xla/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine.
+0: 2023-05-10 00:12:35.493489: I tensorflow/compiler/xla/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine.
+0: 2023-05-10 00:12:35.493493: I tensorflow/compiler/xla/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine.
+0: 2023-05-10 00:12:35.493496: I tensorflow/compiler/xla/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine.
+0: 2023-05-10 00:12:35.493501: I tensorflow/compiler/xla/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine.
+0: 2023-05-10 00:12:57.097945: W tensorflow/compiler/xla/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libnvinfer.so.7'; dlerror: libnvinfer.so.7: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /pfs/lustrep2/projappl/project_462000125/samantao-public/apps-rocm-5.2.3/aws-ofi-rccl:/opt/rocm/lib64:/opt/rocm/lib:/opt/rocm/rocprofiler/lib:/opt/rocm/rocprofiler/tool:/opt/rocm/roctracer/lib:/opt/rocm/roctracer/tool:/opt/rocm/hip/lib:/opt/cray/pe/python/3.9.13.1/lib:/opt/cray/pe/gcc-libs:/opt/cray/libfabric/1.15.2.0/lib64
+0: 2023-05-10 00:12:57.097970: W tensorflow/compiler/xla/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libnvinfer.so.7'; dlerror: libnvinfer.so.7: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /pfs/lustrep2/projappl/project_462000125/samantao-public/apps-rocm-5.2.3/aws-ofi-rccl:/opt/rocm/lib64:/opt/rocm/lib:/opt/rocm/rocprofiler/lib:/opt/rocm/rocprofiler/tool:/opt/rocm/roctracer/lib:/opt/rocm/roctracer/tool:/opt/rocm/hip/lib:/opt/cray/pe/python/3.9.13.1/lib:/opt/cray/pe/gcc-libs:/opt/cray/libfabric/1.15.2.0/lib64
+0: 2023-05-10 00:12:57.097996: W tensorflow/compiler/xla/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libnvinfer.so.7'; dlerror: libnvinfer.so.7: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /pfs/lustrep2/projappl/project_462000125/samantao-public/apps-rocm-5.2.3/aws-ofi-rccl:/opt/rocm/lib64:/opt/rocm/lib:/opt/rocm/rocprofiler/lib:/opt/rocm/rocprofiler/tool:/opt/rocm/roctracer/lib:/opt/rocm/roctracer/tool:/opt/rocm/hip/lib:/opt/cray/pe/python/3.9.13.1/lib:/opt/cray/pe/gcc-libs:/opt/cray/libfabric/1.15.2.0/lib64
+0: 2023-05-10 00:12:57.098006: W tensorflow/compiler/xla/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libnvinfer.so.7'; dlerror: libnvinfer.so.7: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /pfs/lustrep2/projappl/project_462000125/samantao-public/apps-rocm-5.2.3/aws-ofi-rccl:/opt/rocm/lib64:/opt/rocm/lib:/opt/rocm/rocprofiler/lib:/opt/rocm/rocprofiler/tool:/opt/rocm/roctracer/lib:/opt/rocm/roctracer/tool:/opt/rocm/hip/lib:/opt/cray/pe/python/3.9.13.1/lib:/opt/cray/pe/gcc-libs:/opt/cray/libfabric/1.15.2.0/lib64
+0: 2023-05-10 00:12:57.098019: W tensorflow/compiler/xla/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libnvinfer.so.7'; dlerror: libnvinfer.so.7: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /pfs/lustrep2/projappl/project_462000125/samantao-public/apps-rocm-5.2.3/aws-ofi-rccl:/opt/rocm/lib64:/opt/rocm/lib:/opt/rocm/rocprofiler/lib:/opt/rocm/rocprofiler/tool:/opt/rocm/roctracer/lib:/opt/rocm/roctracer/tool:/opt/rocm/hip/lib:/opt/cray/pe/python/3.9.13.1/lib:/opt/cray/pe/gcc-libs:/opt/cray/libfabric/1.15.2.0/lib64
+0: 2023-05-10 00:12:57.098024: W tensorflow/compiler/xla/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libnvinfer.so.7'; dlerror: libnvinfer.so.7: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /pfs/lustrep2/projappl/project_462000125/samantao-public/apps-rocm-5.2.3/aws-ofi-rccl:/opt/rocm/lib64:/opt/rocm/lib:/opt/rocm/rocprofiler/lib:/opt/rocm/rocprofiler/tool:/opt/rocm/roctracer/lib:/opt/rocm/roctracer/tool:/opt/rocm/hip/lib:/opt/cray/pe/python/3.9.13.1/lib:/opt/cray/pe/gcc-libs:/opt/cray/libfabric/1.15.2.0/lib64
+0: 2023-05-10 00:12:57.098033: W tensorflow/compiler/xla/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libnvinfer.so.7'; dlerror: libnvinfer.so.7: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /pfs/lustrep2/projappl/project_462000125/samantao-public/apps-rocm-5.2.3/aws-ofi-rccl:/opt/rocm/lib64:/opt/rocm/lib:/opt/rocm/rocprofiler/lib:/opt/rocm/rocprofiler/tool:/opt/rocm/roctracer/lib:/opt/rocm/roctracer/tool:/opt/rocm/hip/lib:/opt/cray/pe/python/3.9.13.1/lib:/opt/cray/pe/gcc-libs:/opt/cray/libfabric/1.15.2.0/lib64
+0: 2023-05-10 00:12:57.098214: W tensorflow/compiler/xla/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libnvinfer.so.7'; dlerror: libnvinfer.so.7: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /pfs/lustrep2/projappl/project_462000125/samantao-public/apps-rocm-5.2.3/aws-ofi-rccl:/opt/rocm/lib64:/opt/rocm/lib:/opt/rocm/rocprofiler/lib:/opt/rocm/rocprofiler/tool:/opt/rocm/roctracer/lib:/opt/rocm/roctracer/tool:/opt/rocm/hip/lib:/opt/cray/pe/python/3.9.13.1/lib:/opt/cray/pe/gcc-libs:/opt/cray/libfabric/1.15.2.0/lib64
+0: 2023-05-10 00:12:57.118445: W tensorflow/compiler/xla/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libnvinfer_plugin.so.7'; dlerror: libnvinfer_plugin.so.7: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /pfs/lustrep2/projappl/project_462000125/samantao-public/apps-rocm-5.2.3/aws-ofi-rccl:/opt/rocm/lib64:/opt/rocm/lib:/opt/rocm/rocprofiler/lib:/opt/rocm/rocprofiler/tool:/opt/rocm/roctracer/lib:/opt/rocm/roctracer/tool:/opt/rocm/hip/lib:/opt/cray/pe/python/3.9.13.1/lib:/opt/cray/pe/gcc-libs:/opt/cray/libfabric/1.15.2.0/lib64
+0: 2023-05-10 00:12:57.118446: W tensorflow/compiler/xla/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libnvinfer_plugin.so.7'; dlerror: libnvinfer_plugin.so.7: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /pfs/lustrep2/projappl/project_462000125/samantao-public/apps-rocm-5.2.3/aws-ofi-rccl:/opt/rocm/lib64:/opt/rocm/lib:/opt/rocm/rocprofiler/lib:/opt/rocm/rocprofiler/tool:/opt/rocm/roctracer/lib:/opt/rocm/roctracer/tool:/opt/rocm/hip/lib:/opt/cray/pe/python/3.9.13.1/lib:/opt/cray/pe/gcc-libs:/opt/cray/libfabric/1.15.2.0/lib64
+0: 2023-05-10 00:12:57.118459: W tensorflow/compiler/xla/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libnvinfer_plugin.so.7'; dlerror: libnvinfer_plugin.so.7: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /pfs/lustrep2/projappl/project_462000125/samantao-public/apps-rocm-5.2.3/aws-ofi-rccl:/opt/rocm/lib64:/opt/rocm/lib:/opt/rocm/rocprofiler/lib:/opt/rocm/rocprofiler/tool:/opt/rocm/roctracer/lib:/opt/rocm/roctracer/tool:/opt/rocm/hip/lib:/opt/cray/pe/python/3.9.13.1/lib:/opt/cray/pe/gcc-libs:/opt/cray/libfabric/1.15.2.0/lib64
+0: 2023-05-10 00:12:57.118463: W tensorflow/compiler/xla/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libnvinfer_plugin.so.7'; dlerror: libnvinfer_plugin.so.7: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /pfs/lustrep2/projappl/project_462000125/samantao-public/apps-rocm-5.2.3/aws-ofi-rccl:/opt/rocm/lib64:/opt/rocm/lib:/opt/rocm/rocprofiler/lib:/opt/rocm/rocprofiler/tool:/opt/rocm/roctracer/lib:/opt/rocm/roctracer/tool:/opt/rocm/hip/lib:/opt/cray/pe/python/3.9.13.1/lib:/opt/cray/pe/gcc-libs:/opt/cray/libfabric/1.15.2.0/lib64
+0: 2023-05-10 00:12:57.118466: W tensorflow/compiler/xla/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libnvinfer_plugin.so.7'; dlerror: libnvinfer_plugin.so.7: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /pfs/lustrep2/projappl/project_462000125/samantao-public/apps-rocm-5.2.3/aws-ofi-rccl:/opt/rocm/lib64:/opt/rocm/lib:/opt/rocm/rocprofiler/lib:/opt/rocm/rocprofiler/tool:/opt/rocm/roctracer/lib:/opt/rocm/roctracer/tool:/opt/rocm/hip/lib:/opt/cray/pe/python/3.9.13.1/lib:/opt/cray/pe/gcc-libs:/opt/cray/libfabric/1.15.2.0/lib64
+0: 2023-05-10 00:12:57.118480: W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:38] TF-TRT Warning: Cannot dlopen some TensorRT libraries. If you would like to use Nvidia GPU with TensorRT, please make sure the missing libraries mentioned above are installed properly.
+0: 2023-05-10 00:12:57.118479: W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:38] TF-TRT Warning: Cannot dlopen some TensorRT libraries. If you would like to use Nvidia GPU with TensorRT, please make sure the missing libraries mentioned above are installed properly.
+0: 2023-05-10 00:12:57.118472: W tensorflow/compiler/xla/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libnvinfer_plugin.so.7'; dlerror: libnvinfer_plugin.so.7: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /pfs/lustrep2/projappl/project_462000125/samantao-public/apps-rocm-5.2.3/aws-ofi-rccl:/opt/rocm/lib64:/opt/rocm/lib:/opt/rocm/rocprofiler/lib:/opt/rocm/rocprofiler/tool:/opt/rocm/roctracer/lib:/opt/rocm/roctracer/tool:/opt/rocm/hip/lib:/opt/cray/pe/python/3.9.13.1/lib:/opt/cray/pe/gcc-libs:/opt/cray/libfabric/1.15.2.0/lib64
+0: 2023-05-10 00:12:57.118472: W tensorflow/compiler/xla/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libnvinfer_plugin.so.7'; dlerror: libnvinfer_plugin.so.7: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /pfs/lustrep2/projappl/project_462000125/samantao-public/apps-rocm-5.2.3/aws-ofi-rccl:/opt/rocm/lib64:/opt/rocm/lib:/opt/rocm/rocprofiler/lib:/opt/rocm/rocprofiler/tool:/opt/rocm/roctracer/lib:/opt/rocm/roctracer/tool:/opt/rocm/hip/lib:/opt/cray/pe/python/3.9.13.1/lib:/opt/cray/pe/gcc-libs:/opt/cray/libfabric/1.15.2.0/lib64
+0: 2023-05-10 00:12:57.118473: W tensorflow/compiler/xla/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libnvinfer_plugin.so.7'; dlerror: libnvinfer_plugin.so.7: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /pfs/lustrep2/projappl/project_462000125/samantao-public/apps-rocm-5.2.3/aws-ofi-rccl:/opt/rocm/lib64:/opt/rocm/lib:/opt/rocm/rocprofiler/lib:/opt/rocm/rocprofiler/tool:/opt/rocm/roctracer/lib:/opt/rocm/roctracer/tool:/opt/rocm/hip/lib:/opt/cray/pe/python/3.9.13.1/lib:/opt/cray/pe/gcc-libs:/opt/cray/libfabric/1.15.2.0/lib64
+0: 2023-05-10 00:12:57.118504: W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:38] TF-TRT Warning: Cannot dlopen some TensorRT libraries. If you would like to use Nvidia GPU with TensorRT, please make sure the missing libraries mentioned above are installed properly.
+0: 2023-05-10 00:12:57.118507: W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:38] TF-TRT Warning: Cannot dlopen some TensorRT libraries. If you would like to use Nvidia GPU with TensorRT, please make sure the missing libraries mentioned above are installed properly.
+0: 2023-05-10 00:12:57.118503: W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:38] TF-TRT Warning: Cannot dlopen some TensorRT libraries. If you would like to use Nvidia GPU with TensorRT, please make sure the missing libraries mentioned above are installed properly.
+0: 2023-05-10 00:12:57.118515: W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:38] TF-TRT Warning: Cannot dlopen some TensorRT libraries. If you would like to use Nvidia GPU with TensorRT, please make sure the missing libraries mentioned above are installed properly.
+0: 2023-05-10 00:12:57.118516: W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:38] TF-TRT Warning: Cannot dlopen some TensorRT libraries. If you would like to use Nvidia GPU with TensorRT, please make sure the missing libraries mentioned above are installed properly.
+0: 2023-05-10 00:12:57.118517: W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:38] TF-TRT Warning: Cannot dlopen some TensorRT libraries. If you would like to use Nvidia GPU with TensorRT, please make sure the missing libraries mentioned above are installed properly.
+0: Successfully preprocessed all matching files.
+0: Detected CUDA files, patching ldflags
+0: Emitting ninja build file /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/build/build.ninja...
+0: Building extension module scaled_upper_triang_masked_softmax_cuda...
+0: Allowing ninja to set a default number of workers... (overridable by setting the environment variable MAX_JOBS=N)
+0: Loading extension module scaled_upper_triang_masked_softmax_cuda...
+0: Successfully preprocessed all matching files.
+0: Detected CUDA files, patching ldflags
+0: Emitting ninja build file /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/build/build.ninja...
+0: Building extension module scaled_masked_softmax_cuda...
+0: Allowing ninja to set a default number of workers... (overridable by setting the environment variable MAX_JOBS=N)
+0: Loading extension module scaled_masked_softmax_cuda...
+0: Successfully preprocessed all matching files.
+0: Detected CUDA files, patching ldflags
+0: Emitting ninja build file /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/build/build.ninja...
+0: Building extension module fused_mix_prec_layer_norm_cuda...
+0: Allowing ninja to set a default number of workers... (overridable by setting the environment variable MAX_JOBS=N)
+0: Loading extension module fused_mix_prec_layer_norm_cuda...
+0: Successfully preprocessed all matching files.
+0: Successfully preprocessed all matching files.
+0: Successfully preprocessed all matching files.
+0: Successfully preprocessed all matching files.
+0: Successfully preprocessed all matching files.
+0: Successfully preprocessed all matching files.
+0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/venv/lib/python3.9/site-packages/torch/distributed/distributed_c10d.py:429: UserWarning: torch.distributed.distributed_c10d._get_global_rank is deprecated please use torch.distributed.distributed_c10d.get_global_rank instead
+0:   warnings.warn(
+0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/venv/lib/python3.9/site-packages/torch/distributed/distributed_c10d.py:429: UserWarning: torch.distributed.distributed_c10d._get_global_rank is deprecated please use torch.distributed.distributed_c10d.get_global_rank instead
+0:   warnings.warn(
+0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/venv/lib/python3.9/site-packages/torch/distributed/distributed_c10d.py:429: UserWarning: torch.distributed.distributed_c10d._get_global_rank is deprecated please use torch.distributed.distributed_c10d.get_global_rank instead
+0:   warnings.warn(
+0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/venv/lib/python3.9/site-packages/torch/distributed/distributed_c10d.py:429: UserWarning: torch.distributed.distributed_c10d._get_global_rank is deprecated please use torch.distributed.distributed_c10d.get_global_rank instead
+0:   warnings.warn(
+0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/venv/lib/python3.9/site-packages/torch/distributed/distributed_c10d.py:429: UserWarning: torch.distributed.distributed_c10d._get_global_rank is deprecated please use torch.distributed.distributed_c10d.get_global_rank instead
+0:   warnings.warn(
+0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/venv/lib/python3.9/site-packages/torch/distributed/distributed_c10d.py:429: UserWarning: torch.distributed.distributed_c10d._get_global_rank is deprecated please use torch.distributed.distributed_c10d.get_global_rank instead
+0:   warnings.warn(
+0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/venv/lib/python3.9/site-packages/torch/distributed/distributed_c10d.py:429: UserWarning: torch.distributed.distributed_c10d._get_global_rank is deprecated please use torch.distributed.distributed_c10d.get_global_rank instead
+0:   warnings.warn(
+0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/venv/lib/python3.9/site-packages/torch/distributed/distributed_c10d.py:429: UserWarning: torch.distributed.distributed_c10d._get_global_rank is deprecated please use torch.distributed.distributed_c10d.get_global_rank instead
+0:   warnings.warn(
+0: Using /pfs/lustrep4/users/muennighoff/.cache/torch_extensions/py39_cpu as PyTorch extensions root...Using /pfs/lustrep4/users/muennighoff/.cache/torch_extensions/py39_cpu as PyTorch extensions root...Using /pfs/lustrep4/users/muennighoff/.cache/torch_extensions/py39_cpu as PyTorch extensions root...Using /pfs/lustrep4/users/muennighoff/.cache/torch_extensions/py39_cpu as PyTorch extensions root...Using /pfs/lustrep4/users/muennighoff/.cache/torch_extensions/py39_cpu as PyTorch extensions root...
+0: 
+0: 
+0: Using /pfs/lustrep4/users/muennighoff/.cache/torch_extensions/py39_cpu as PyTorch extensions root...
+0: 
+0: Using /pfs/lustrep4/users/muennighoff/.cache/torch_extensions/py39_cpu as PyTorch extensions root...
+0: 
+0: Using /pfs/lustrep4/users/muennighoff/.cache/torch_extensions/py39_cpu as PyTorch extensions root...
+0: Emitting ninja build file /pfs/lustrep4/users/muennighoff/.cache/torch_extensions/py39_cpu/utils/build.ninja...
+0: Building extension module utils...
+0: Allowing ninja to set a default number of workers... (overridable by setting the environment variable MAX_JOBS=N)
+0: Loading extension module utils...
+0: Loading extension module utils...
+0: Loading extension module utils...
+0: Loading extension module utils...
+0: Loading extension module utils...
+0: Loading extension module utils...
+0: Loading extension module utils...
+0: Loading extension module utils...
+0: Using /pfs/lustrep4/users/muennighoff/.cache/torch_extensions/py39_cpu as PyTorch extensions root...Using /pfs/lustrep4/users/muennighoff/.cache/torch_extensions/py39_cpu as PyTorch extensions root...
+0: 
+0: Using /pfs/lustrep4/users/muennighoff/.cache/torch_extensions/py39_cpu as PyTorch extensions root...
+0: Using /pfs/lustrep4/users/muennighoff/.cache/torch_extensions/py39_cpu as PyTorch extensions root...
+0: Using /pfs/lustrep4/users/muennighoff/.cache/torch_extensions/py39_cpu as PyTorch extensions root...
+0: Using /pfs/lustrep4/users/muennighoff/.cache/torch_extensions/py39_cpu as PyTorch extensions root...
+0: Using /pfs/lustrep4/users/muennighoff/.cache/torch_extensions/py39_cpu as PyTorch extensions root...
+0: No modifications detected for re-loaded extension module utils, skipping build step...No modifications detected for re-loaded extension module utils, skipping build step...
+0: 
+0: Loading extension module utils...No modifications detected for re-loaded extension module utils, skipping build step...Loading extension module utils...
+0: 
+0: 
+0: Loading extension module utils...
+0: No modifications detected for re-loaded extension module utils, skipping build step...
+0: Loading extension module utils...
+0: No modifications detected for re-loaded extension module utils, skipping build step...
+0: Loading extension module utils...
+0: No modifications detected for re-loaded extension module utils, skipping build step...
+0: Loading extension module utils...
+0: No modifications detected for re-loaded extension module utils, skipping build step...
+0: Loading extension module utils...
+0: Using /pfs/lustrep4/users/muennighoff/.cache/torch_extensions/py39_cpu as PyTorch extensions root...
+0: No modifications detected for re-loaded extension module utils, skipping build step...
+0: Loading extension module utils...
+0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/utils.py:349: UserWarning: Parameter count with the embeddings will be inaccurate with PP > 1, as the first and last stage hold several copies of the embeddings
+0:   warnings.warn("Parameter count with the embeddings will be inaccurate with PP > 1, as the first and last stage hold several copies of the embeddings")

21m400m400m/3487337.out

@@ -0,0 +1,852 @@
+Model parameters: d_model 288 ffw_size 1152 kv_size 32 n_heads 7 n_layers 5
+Megatron-DeepSpeed/pretrain_gpt.py --tensor-model-parallel-size 1 --pipeline-model-parallel-size 1 --num-layers 5 --hidden-size 288 --num-attention-heads 7 --kv-channels 32 --ffn-hidden-size 1152 --seq-length 2048 --max-position-embeddings 2048 --micro-batch-size 32 --global-batch-size 256 --train-samples 1 --vocab-file gpt2/vocab.json --merge-file gpt2/merges.txt --clip-grad 1.0 --kill-switch-path kill-switch-21m400m400mval --bf16 --optimizer adam --adam-beta1 0.9 --adam-beta2 0.999 --adam-eps 1e-8 --lr 2e-4 --min-lr 2e-5 --lr-decay-style cosine --lr-decay-samples 1 --lr-warmup-samples 0 --clip-grad 1.0 --weight-decay 1e-1 --override-lr-scheduler --no-load-optim --reset-progress --log-interval 10 --save-interval 1000 --eval-interval 1 --eval-iters 100 --tensorboard-dir tensorboard_21m400m400mval --tensorboard-queue-size 5 --log-timers-to-tensorboard --log-batch-size-to-tensorboard --log-validation-ppl-to-tensorboard --save checkpoints_21m400m400m --load checkpoints_21m400m400m --train-weighted-split-paths-path train400m.txt --valid-weighted-split-paths-path val.txt --data-impl mmap --deepspeed --deepspeed_config ds_configs/3487337.json --zero-stage 0
+START 3487337: Wed 10 May 2023 12:11:40 AM EEST
+0: 
+0: 
+0: ======================= ROCm System Management Interface =======================
+0: ================================= Concise Info =================================
+0: GPU  Temp   AvgPwr  SCLK    MCLK     Fan  Perf  PwrCap  VRAM%  GPU%  
+0: 0    43.0c  92.0W   800Mhz  1600Mhz  0%   auto  560.0W    0%   0%    
+0: 1    48.0c  N/A     800Mhz  1600Mhz  0%   auto  0.0W      0%   0%    
+0: 2    44.0c  90.0W   800Mhz  1600Mhz  0%   auto  560.0W    0%   0%    
+0: 3    44.0c  N/A     800Mhz  1600Mhz  0%   auto  0.0W      0%   0%    
+0: 4    46.0c  91.0W   800Mhz  1600Mhz  0%   auto  560.0W    0%   0%    
+0: 5    42.0c  N/A     800Mhz  1600Mhz  0%   auto  0.0W      0%   0%    
+0: 6    43.0c  87.0W   800Mhz  1600Mhz  0%   auto  560.0W    0%   0%    
+0: 7    46.0c  N/A     800Mhz  1600Mhz  0%   auto  0.0W      0%   0%    
+0: ================================================================================
+0: ============================= End of ROCm SMI Log ==============================
+0: Launching on nid007269 (0/1), master nid007269 port 9999, GPUs 8, CUDA: True
+0: using world size: 8, data-parallel-size: 8, tensor-model-parallel size: 1, pipeline-model-parallel size: 1 
+0: accumulate and all-reduce gradients in fp32 for bfloat16 data type.
+0: using torch.bfloat16 for parameters ...
+0: ------------------------ arguments ------------------------
+0:   abort_on_unmet_fused_kernel_constraints ......... False
+0:   accumulate_allreduce_grads_in_fp32 .............. True
+0:   adam_beta1 ...................................... 0.9
+0:   adam_beta2 ...................................... 0.999
+0:   adam_eps ........................................ 1e-08
+0:   adlr_autoresume ................................. False
+0:   adlr_autoresume_interval ........................ 1000
+0:   apply_query_key_layer_scaling ................... True
+0:   apply_residual_connection_post_layernorm ........ False
+0:   attention_dropout ............................... 0.1
+0:   attention_softmax_in_fp32 ....................... False
+0:   bert_binary_head ................................ True
+0:   bert_load ....................................... None
+0:   bf16 ............................................ True
+0:   bias_dropout_fusion ............................. True
+0:   bias_gelu_fusion ................................ True
+0:   biencoder_projection_dim ........................ 0
+0:   biencoder_shared_query_context_model ............ False
+0:   block_data_path ................................. None
+0:   checkpoint_activations .......................... False
+0:   checkpoint_in_cpu ............................... False
+0:   checkpoint_num_layers ........................... 1
+0:   clip_grad ....................................... 1.0
+0:   codecarbon_dir .................................. None
+0:   consumed_train_samples .......................... 0
+0:   consumed_train_tokens ........................... 0
+0:   consumed_valid_samples .......................... 0
+0:   contigious_checkpointing ........................ False
+0:   cpu_optimizer ................................... False
+0:   cpu_torch_adam .................................. False
+0:   curriculum_learning ............................. False
+0:   data_impl ....................................... mmap
+0:   data_parallel_size .............................. 8
+0:   data_path ....................................... None
+0:   dataloader_type ................................. single
+0:   DDP_impl ........................................ local
+0:   decoder_seq_length .............................. None
+0:   deepscale ....................................... False
+0:   deepscale_config ................................ None
+0:   deepspeed ....................................... True
+0:   deepspeed_activation_checkpointing .............. False
+0:   deepspeed_config ................................ ds_configs/3487337.json
+0:   deepspeed_mpi ................................... False
+0:   distribute_checkpointed_activations ............. False
+0:   distributed_backend ............................. nccl
+0:   embed_layernorm ................................. False
+0:   embedding_path .................................. None
+0:   encoder_seq_length .............................. 2048
+0:   eod_mask_loss ................................... False
+0:   eval_interval ................................... 1
+0:   eval_iters ...................................... 100
+0:   eval_only ....................................... None
+0:   evidence_data_path .............................. None
+0:   exit_duration_in_mins ........................... None
+0:   exit_interval ................................... None
+0:   ffn_hidden_size ................................. 1152
+0:   finetune ........................................ False
+0:   fp16 ............................................ False
+0:   fp16_lm_cross_entropy ........................... False
+0:   fp32_residual_connection ........................ False
+0:   gigaflos_no_embeds .............................. 0
+0:   global_batch_size ............................... 256
+0:   glu_activation .................................. None
+0:   hidden_dropout .................................. 0.1
+0:   hidden_size ..................................... 288
+0:   hysteresis ...................................... 2
+0:   ict_head_size ................................... None
+0:   ict_load ........................................ None
+0:   img_dim ......................................... 224
+0:   indexer_batch_size .............................. 128
+0:   indexer_log_interval ............................ 1000
+0:   inference ....................................... False
+0:   init_method_std ................................. 0.02
+0:   init_method_xavier_uniform ...................... False
+0:   initial_loss_scale .............................. 4294967296
+0:   kill_switch_path ................................ kill-switch-21m400m400mval
+0:   kv_channels ..................................... 32
+0:   layer_norm_fusion ............................... True
+0:   layernorm_epsilon ............................... 1e-05
+0:   lazy_mpu_init ................................... None
+0:   load ............................................ checkpoints_21m400m400m
+0:   local_rank ...................................... None
+0:   log_batch_size_to_tensorboard ................... True
+0:   log_interval .................................... 10
+0:   log_learning_rate_to_tensorboard ................ True
+0:   log_level ....................................... None
+0:   log_level_replica ............................... None
+0:   log_loss_scale_to_tensorboard ................... True
+0:   log_num_zeros_in_grad ........................... False
+0:   log_params_norm ................................. False
+0:   log_path ........................................ None
+0:   log_timers_to_tensorboard ....................... True
+0:   log_validation_ppl_to_tensorboard ............... True
+0:   loss_on_targets_only ............................ False
+0:   loss_scale ...................................... None
+0:   loss_scale_window ............................... 1000
+0:   lr .............................................. 0.0002
+0:   lr_decay_iters .................................. None
+0:   lr_decay_samples ................................ 1
+0:   lr_decay_style .................................. cosine
+0:   lr_decay_tokens ................................. None
+0:   lr_warmup_fraction .............................. None
+0:   lr_warmup_iters ................................. 0
+0:   lr_warmup_samples ............................... 0
+0:   make_vocab_size_divisible_by .................... 128
+0:   mask_prob ....................................... 0.15
+0:   masked_softmax_fusion ........................... True
+0:   max_position_embeddings ......................... 2048
+0:   mean_noise_span_length .......................... None
+0:   memory_centric_tiled_linear ..................... False
+0:   merge_file ...................................... gpt2/merges.txt
+0:   micro_batch_size ................................ 32
+0:   min_loss_scale .................................. 1.0
+0:   min_lr .......................................... 2e-05
+0:   mmap_warmup ..................................... False
+0:   no_load_optim ................................... True
+0:   no_load_rng ..................................... None
+0:   no_save_optim ................................... None
+0:   no_save_rng ..................................... None
+0:   noise_density ................................... None
+0:   num_attention_heads ............................. 7
+0:   num_channels .................................... 3
+0:   num_classes ..................................... 1000
+0:   num_layers ...................................... 5
+0:   num_layers_per_virtual_pipeline_stage ........... None
+0:   num_workers ..................................... 2
+0:   onnx_safe ....................................... None
+0:   openai_gelu ..................................... False
+0:   optimizer ....................................... adam
+0:   optimizer_fusion ................................ True
+0:   override_lr_scheduler ........................... True
+0:   pad_vocab_size_to ............................... None
+0:   params_dtype .................................... torch.bfloat16
+0:   partition_activations ........................... False
+0:   patch_dim ....................................... 16
+0:   pipeline_model_parallel_size .................... 1
+0:   position_embedding_type ......................... PositionEmbeddingType.absolute
+0:   pp_partition_method ............................. None
+0:   profile_backward ................................ False
+0:   query_in_block_prob ............................. 0.1
+0:   rampup_batch_size ............................... None
+0:   rank ............................................ 0
+0:   remote_device ................................... none
+0:   reset_attention_mask ............................ False
+0:   reset_position_ids .............................. False
+0:   reset_progress .................................. True
+0:   retriever_report_topk_accuracies ................ []
+0:   retriever_score_scaling ......................... False
+0:   retriever_seq_length ............................ 256
+0:   reweight_loss_based_on_position_frequency ....... False
+0:   sample_rate ..................................... 1.0
+0:   save ............................................ checkpoints_21m400m400m
+0:   save_interval ................................... 1000
+0:   scatter_gather_tensors_in_pipeline .............. True
+0:   scattered_embeddings ............................ False
+0:   seed ............................................ 1234
+0:   seq_length ...................................... 2048
+0:   sgd_momentum .................................... 0.9
+0:   short_seq_prob .................................. 0.1
+0:   skip_train_iteration_range ...................... None
+0:   split ........................................... None
+0:   split_transformers .............................. False
+0:   sync_tp_duplicated_parameters ................... False
+0:   synchronize_each_layer .......................... False
+0:   tensor_model_parallel_size ...................... 1
+0:   tensorboard_dir ................................. tensorboard_21m400m400mval
+0:   tensorboard_log_interval ........................ 1
+0:   tensorboard_queue_size .......................... 5
+0:   test_weighted_split_paths ....................... None
+0:   test_weighted_split_paths_path .................. None
+0:   tile_factor ..................................... 1
+0:   titles_data_path ................................ None
+0:   tokenizer_name_or_path .......................... None
+0:   tokenizer_type .................................. GPT2BPETokenizer
+0:   train_iters ..................................... None
+0:   train_samples ................................... 1
+0:   train_tokens .................................... None
+0:   train_weighted_split_names ...................... ['train']
+0:   train_weighted_split_paths ...................... [['/scratch/project_462000119/data/c4_subsampled/gpt2tok_c4_en_400M_text_document']]
+0:   train_weighted_split_paths_path ................. None
+0:   train_weighted_split_splits ..................... [['0:1']]
+0:   train_weighted_split_weights .................... [['1.0']]
+0:   universal_checkpoint ............................ False
+0:   use_bnb_optimizer ............................... False
+0:   use_checkpoint_lr_scheduler ..................... False
+0:   use_contiguous_buffers_in_ddp ................... True
+0:   use_cpu_initialization .......................... None
+0:   use_one_sent_docs ............................... False
+0:   use_pin_memory .................................. False
+0:   valid_num_workers ............................... 2
+0:   valid_weighted_split_names ...................... ['validation']
+0:   valid_weighted_split_paths ...................... [['/scratch/project_462000119/data/c4_validation/gpt2tok_c4validation_rerun_text_document']]
+0:   valid_weighted_split_paths_path ................. None
+0:   valid_weighted_split_splits ..................... [['0:1']]
+0:   valid_weighted_split_weights .................... [['1.0']]
+0:   virtual_pipeline_model_parallel_size ............ None
+0:   vocab_extra_ids ................................. 0
+0:   vocab_file ...................................... gpt2/vocab.json
+0:   weight_decay .................................... 0.1
+0:   world_size ...................................... 8
+0:   zero_allgather_bucket_size ...................... 0.0
+0:   zero_contigious_gradients ....................... False
+0:   zero_reduce_bucket_size ......................... 0.0
+0:   zero_reduce_scatter ............................. False
+0:   zero_stage ...................................... 0
+0: -------------------- end of arguments ---------------------
+0: setting number of micro-batches to constant 1
+0: > building GPT2BPETokenizer tokenizer ...
+0:  > padded vocab (size: 50257) with 47 dummy tokens (new size: 50304)
+0: DeepSpeed general environment info:
+0: torch install path ............... ['/pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/venv/lib/python3.9/site-packages/torch']
+0: torch version .................... 1.13.0+rocm5.2
+0: torch cuda version ............... None
+0: torch hip version ................ 5.2.21151-afdc89f8
+0: nvcc version ..................... None
+0: deepspeed install path ........... ['/pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/venv/lib/python3.9/site-packages/deepspeed']
+0: deepspeed info ................... 0.7.5, unknown, unknown
+0: deepspeed wheel compiled w. ...... torch 1.13, hip 5.1
+0: **** Git info for Megatron: git_hash=unknown git_branch=unknown ****
+0: > initializing torch distributed ...
+0: [2023-05-10 00:13:50,262] [INFO] [comm.py:633:init_distributed] Initializing TorchBackend in DeepSpeed with backend nccl
+0: > setting tensorboard ...
+0: > initializing tensor model parallel with size 1
+0: > initializing pipeline model parallel with size 1
+0: > setting random seeds to 1234 ...
+0: > initializing model parallel cuda seeds on global rank 0, model parallel rank 0, and data parallel rank 0 with model parallel seed: 3952 and data parallel seed: 1234
+0: > compiling dataset index builder ...
+0: make: Entering directory '/pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/data'
+0: make: Nothing to be done for 'default'.
+0: make: Leaving directory '/pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/data'
+0: >>> done with dataset index builder. Compilation time: 0.118 seconds
+0: > compiling and loading fused kernels ...
+0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_upper_triang_masked_softmax.cpp -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_upper_triang_masked_softmax_hip.cpp [skipped, already hipified]
+0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_upper_triang_masked_softmax.h -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_upper_triang_masked_softmax_hip.h [skipped, already hipified]
+0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/compat.h -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/compat.h [skipped, no changes]
+0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/type_shim.h -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/type_shim.h [skipped, no changes]
+0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_upper_triang_masked_softmax_cuda.cu -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_upper_triang_masked_softmax_hip.hip [skipped, already hipified]
+0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/type_shim.h -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/type_shim.h [skipped, no changes]
+0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/compat.h -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/compat.h [skipped, no changes]
+0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_upper_triang_masked_softmax.h -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_upper_triang_masked_softmax_hip.h [skipped, already hipified]
+0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_masked_softmax.h -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_masked_softmax_hip.h [skipped, already hipified]
+0: Total number of unsupported CUDA function calls: 0
+0: 
+0: 
+0: Total number of replaced kernel launches: 87
+0: ninja: no work to do.
+0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_masked_softmax.cpp -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_masked_softmax_hip.cpp [skipped, already hipified]
+0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_masked_softmax_cuda.cu -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_masked_softmax_hip.hip [skipped, already hipified]
+0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/type_shim.h -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/type_shim.h [skipped, no changes]
+0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/compat.h -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/compat.h [skipped, no changes]
+0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_upper_triang_masked_softmax.h -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_upper_triang_masked_softmax_hip.h [skipped, already hipified]
+0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_masked_softmax.h -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_masked_softmax_hip.h [skipped, already hipified]
+0: Total number of unsupported CUDA function calls: 0
+0: 
+0: 
+0: Total number of replaced kernel launches: 63
+0: ninja: no work to do.
+0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/layer_norm_cuda.cpp -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/layer_norm_cuda.cpp [skipped, no changes]
+0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/layer_norm_cuda_kernel.cu -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/layer_norm_hip_kernel.hip [skipped, already hipified]
+0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/type_shim.h -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/type_shim.h [skipped, no changes]
+0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/compat.h -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/compat.h [skipped, no changes]
+0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_upper_triang_masked_softmax.h -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_upper_triang_masked_softmax_hip.h [skipped, already hipified]
+0: /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_masked_softmax.h -> /pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/Megatron-DeepSpeed/megatron/fused_kernels/scaled_masked_softmax_hip.h [skipped, already hipified]
+0: Total number of unsupported CUDA function calls: 0
+0: 
+0: 
+0: Total number of replaced kernel launches: 67
+0: [1/1] c++ layer_norm_cuda.o layer_norm_hip_kernel.cuda.o -shared -L/pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/venv/lib/python3.9/site-packages/torch/lib -lc10 -lc10_hip -ltorch_cpu -ltorch_hip -ltorch -ltorch_python -L/opt/rocm/lib -lamdhip64 -o fused_mix_prec_layer_norm_cuda.so
+0: >>> done with compiling and loading fused kernels. Compilation time: 11.020 seconds
+0: time to initialize megatron (seconds): 26.195
+0: [after megatron is initialized] datetime: 2023-05-10 00:14:02 
+0: building GPT model ...
+0: [2023-05-10 00:14:02,324] [INFO] [utils.py:827:see_memory_usage] Before Building Model
+0: [2023-05-10 00:14:02,325] [INFO] [utils.py:828:see_memory_usage] MA 0.0 GB         Max_MA 0.0 GB         CA 0.0 GB         Max_CA 0 GB 
+0: [2023-05-10 00:14:02,325] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory:  used = 37.4 GB, percent = 7.4%
+0: SEED_LAYERS=False BASE_SEED=1234 SEED_FN=None
+0: Using topology: {ProcessCoord(pipe=0, data=0, model=0): 0, ProcessCoord(pipe=0, data=1, model=0): 1, ProcessCoord(pipe=0, data=2, model=0): 2, ProcessCoord(pipe=0, data=3, model=0): 3, ProcessCoord(pipe=0, data=4, model=0): 4, ProcessCoord(pipe=0, data=5, model=0): 5, ProcessCoord(pipe=0, data=6, model=0): 6, ProcessCoord(pipe=0, data=7, model=0): 7}
+0: [2023-05-10 00:14:02,576] [INFO] [module.py:366:_partition_layers] Partitioning pipeline stages with method type:transformer
+0: stage=0 layers=12
+0:      0: _to_float16
+0:      1: EmbeddingPipe
+0:      2: <lambda>
+0:      3: ParallelTransformerLayerPipe
+0:      4: ParallelTransformerLayerPipe
+0:      5: ParallelTransformerLayerPipe
+0:      6: ParallelTransformerLayerPipe
+0:      7: ParallelTransformerLayerPipe
+0:      8: undo
+0:      9: MixedFusedLayerNorm
+0:     10: EmbeddingPipe
+0:     11: float16_to_fp32
+0:   loss: CrossEntropy
+0: [2023-05-10 00:14:03,105] [INFO] [utils.py:827:see_memory_usage] After Building Model
+0: [2023-05-10 00:14:03,106] [INFO] [utils.py:828:see_memory_usage] MA 0.04 GB         Max_MA 0.04 GB         CA 0.06 GB         Max_CA 0 GB 
+0: [2023-05-10 00:14:03,106] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory:  used = 37.41 GB, percent = 7.4%
+0: setting training iterations to 0
+0: > learning rate decay style: cosine
+0: DeepSpeed is enabled.
+0: [2023-05-10 00:14:03,107] [INFO] [logging.py:68:log_dist] [Rank 0] DeepSpeed info: version=0.7.5, git-hash=unknown, git-branch=unknown
+0: [2023-05-10 00:14:08,294] [INFO] [logging.py:68:log_dist] [Rank 0] DeepSpeed Flops Profiler Enabled: False
+0: [2023-05-10 00:14:08,295] [INFO] [logging.py:68:log_dist] [Rank 0] Removing param_group that has no 'params' in the client Optimizer
+0: [2023-05-10 00:14:08,295] [INFO] [logging.py:68:log_dist] [Rank 0] Using client Optimizer as basic optimizer
+0: [2023-05-10 00:14:08,296] [INFO] [logging.py:68:log_dist] [Rank 0] DeepSpeed Basic Optimizer = FusedAdam
+0: [2023-05-10 00:14:08,296] [INFO] [logging.py:68:log_dist] [Rank 0] Creating BF16 optimizer
+0: [2023-05-10 00:14:08,414] [INFO] [utils.py:827:see_memory_usage] begin bf16_optimizer
+0: [2023-05-10 00:14:08,415] [INFO] [utils.py:828:see_memory_usage] MA 0.04 GB         Max_MA 0.04 GB         CA 0.06 GB         Max_CA 0 GB 
+0: [2023-05-10 00:14:08,415] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory:  used = 38.96 GB, percent = 7.7%
+0: ninja: no work to do.
+0: Time to load utils op: 0.7600808143615723 seconds
+0: Time to load utils op: 0.7584981918334961 seconds
+0: Time to load utils op: 0.7595300674438477 seconds
+0: Time to load utils op: 0.7585179805755615 seconds
+0: Time to load utils op: 0.7589340209960938 seconds
+0: Time to load utils op: 0.7584795951843262 seconds
+0: Time to load utils op: 0.6399149894714355 seconds
+0: Time to load utils op: 0.760486364364624 seconds
+0: [2023-05-10 00:14:09,166] [INFO] [utils.py:827:see_memory_usage] before initializing group 0
+0: [2023-05-10 00:14:09,167] [INFO] [utils.py:828:see_memory_usage] MA 0.04 GB         Max_MA 0.04 GB         CA 0.06 GB         Max_CA 0 GB 
+0: [2023-05-10 00:14:09,167] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory:  used = 38.29 GB, percent = 7.6%
+0: Time to load utils op: 0.0005950927734375 secondsTime to load utils op: 0.00048732757568359375 seconds
+0: 
+0: Time to load utils op: 0.00044083595275878906 seconds
+0: Time to load utils op: 0.0006310939788818359 seconds
+0: Time to load utils op: 0.0004439353942871094 seconds
+0: Time to load utils op: 0.0006113052368164062 seconds
+0: Time to load utils op: 0.0011534690856933594 seconds
+0: [2023-05-10 00:14:10,015] [INFO] [utils.py:827:see_memory_usage] after initializing group 0
+0: [2023-05-10 00:14:10,016] [INFO] [utils.py:828:see_memory_usage] MA 0.11 GB         Max_MA 0.11 GB         CA 0.15 GB         Max_CA 0 GB 
+0: [2023-05-10 00:14:10,016] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory:  used = 37.55 GB, percent = 7.5%
+0: [2023-05-10 00:14:10,123] [INFO] [utils.py:827:see_memory_usage] before initializing group 1
+0: [2023-05-10 00:14:10,124] [INFO] [utils.py:828:see_memory_usage] MA 0.11 GB         Max_MA 0.11 GB         CA 0.15 GB         Max_CA 0 GB 
+0: [2023-05-10 00:14:10,124] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory:  used = 37.53 GB, percent = 7.5%
+0: [2023-05-10 00:14:10,226] [INFO] [utils.py:827:see_memory_usage] after initializing group 1
+0: [2023-05-10 00:14:10,226] [INFO] [utils.py:828:see_memory_usage] MA 0.12 GB         Max_MA 0.12 GB         CA 0.15 GB         Max_CA 0 GB 
+0: [2023-05-10 00:14:10,227] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory:  used = 37.53 GB, percent = 7.5%
+0: [2023-05-10 00:14:10,327] [INFO] [utils.py:827:see_memory_usage] before initializing group 2
+0: [2023-05-10 00:14:10,328] [INFO] [utils.py:828:see_memory_usage] MA 0.12 GB         Max_MA 0.12 GB         CA 0.15 GB         Max_CA 0 GB 
+0: [2023-05-10 00:14:10,328] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory:  used = 37.53 GB, percent = 7.5%
+0: [2023-05-10 00:14:10,430] [INFO] [utils.py:827:see_memory_usage] after initializing group 2
+0: [2023-05-10 00:14:10,430] [INFO] [utils.py:828:see_memory_usage] MA 0.12 GB         Max_MA 0.12 GB         CA 0.15 GB         Max_CA 0 GB 
+0: [2023-05-10 00:14:10,430] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory:  used = 37.53 GB, percent = 7.5%
+0: [2023-05-10 00:14:10,531] [INFO] [utils.py:827:see_memory_usage] before initialize_optimizer
+0: [2023-05-10 00:14:10,531] [INFO] [utils.py:828:see_memory_usage] MA 0.12 GB         Max_MA 0.12 GB         CA 0.15 GB         Max_CA 0 GB 
+0: [2023-05-10 00:14:10,531] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory:  used = 37.53 GB, percent = 7.5%
+0: [2023-05-10 00:14:10,637] [INFO] [utils.py:827:see_memory_usage] end initialize_optimizer
+0: [2023-05-10 00:14:10,638] [INFO] [utils.py:828:see_memory_usage] MA 0.14 GB         Max_MA 0.14 GB         CA 0.15 GB         Max_CA 0 GB 
+0: [2023-05-10 00:14:10,638] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory:  used = 37.53 GB, percent = 7.5%
+0: [2023-05-10 00:14:10,738] [INFO] [utils.py:827:see_memory_usage] end bf16_optimizer
+0: [2023-05-10 00:14:10,739] [INFO] [utils.py:828:see_memory_usage] MA 0.14 GB         Max_MA 0.14 GB         CA 0.15 GB         Max_CA 0 GB 
+0: [2023-05-10 00:14:10,739] [INFO] [utils.py:836:see_memory_usage] CPU Virtual Memory:  used = 37.53 GB, percent = 7.5%
+0: [2023-05-10 00:14:10,739] [INFO] [logging.py:68:log_dist] [Rank 0] DeepSpeed Final Optimizer = FusedAdam
+0: [2023-05-10 00:14:10,739] [INFO] [logging.py:68:log_dist] [Rank 0] DeepSpeed using client LR scheduler
+0: [2023-05-10 00:14:10,739] [INFO] [logging.py:68:log_dist] [Rank 0] DeepSpeed LR Scheduler = <megatron.learning_rates.AnnealingLR object at 0x14b88d151fd0>
+0: [2023-05-10 00:14:10,739] [INFO] [logging.py:68:log_dist] [Rank 0] step=0, skipped=0, lr=[0.0002, 0.0002, 0.0002], mom=[(0.9, 0.999), (0.9, 0.999), (0.9, 0.999)]
+0: [2023-05-10 00:14:10,739] [INFO] [config.py:1007:print] DeepSpeedEngine configuration:
+0: [2023-05-10 00:14:10,740] [INFO] [config.py:1011:print]   activation_checkpointing_config  {
+0:     "partition_activations": false, 
+0:     "contiguous_memory_optimization": false, 
+0:     "cpu_checkpointing": false, 
+0:     "number_checkpoints": null, 
+0:     "synchronize_checkpoint_boundary": false, 
+0:     "profile": false
+0: }
+0: [2023-05-10 00:14:10,740] [INFO] [config.py:1011:print]   aio_config ................... {'block_size': 1048576, 'queue_depth': 8, 'thread_count': 1, 'single_submit': False, 'overlap_events': True}
+0: [2023-05-10 00:14:10,740] [INFO] [config.py:1011:print]   amp_enabled .................. False
+0: [2023-05-10 00:14:10,740] [INFO] [config.py:1011:print]   amp_params ................... False
+0: [2023-05-10 00:14:10,740] [INFO] [config.py:1011:print]   autotuning_config ............ {
+0:     "enabled": false, 
+0:     "start_step": null, 
+0:     "end_step": null, 
+0:     "metric_path": null, 
+0:     "arg_mappings": null, 
+0:     "metric": "throughput", 
+0:     "model_info": null, 
+0:     "results_dir": "/pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/autotuning_results", 
+0:     "exps_dir": "/pfs/lustrep4/scratch/project_462000119/muennighoff/nov-2022-bettercom/autotuning_exps", 
+0:     "overwrite": true, 
+0:     "fast": true, 
+0:     "start_profile_step": 3, 
+0:     "end_profile_step": 5, 
+0:     "tuner_type": "gridsearch", 
+0:     "tuner_early_stopping": 5, 
+0:     "tuner_num_trials": 50, 
+0:     "model_info_path": null, 
+0:     "mp_size": 1, 
+0:     "max_train_batch_size": null, 
+0:     "min_train_batch_size": 1, 
+0:     "max_train_micro_batch_size_per_gpu": 1.024000e+03, 
+0:     "min_train_micro_batch_size_per_gpu": 1, 
+0:     "num_tuning_micro_batch_sizes": 3
+0: }
+0: [2023-05-10 00:14:10,740] [INFO] [config.py:1011:print]   bfloat16_enabled ............. True
+0: [2023-05-10 00:14:10,740] [INFO] [config.py:1011:print]   checkpoint_parallel_write_pipeline  False
+0: [2023-05-10 00:14:10,740] [INFO] [config.py:1011:print]   checkpoint_tag_validation_enabled  True
+0: [2023-05-10 00:14:10,740] [INFO] [config.py:1011:print]   checkpoint_tag_validation_fail  False
+0: [2023-05-10 00:14:10,740] [INFO] [config.py:1011:print]   comms_config ................. <deepspeed.comm.config.DeepSpeedCommsConfig object at 0x14b88d151d60>
+0: [2023-05-10 00:14:10,740] [INFO] [config.py:1011:print]   communication_data_type ...... None
+0: [2023-05-10 00:14:10,740] [INFO] [config.py:1011:print]   compression_config ........... {'weight_quantization': {'shared_parameters': {'enabled': False, 'quantizer_kernel': False, 'schedule_offset': 0, 'quantize_groups': 1, 'quantize_verbose': False, 'quantization_type': 'symmetric', 'quantize_weight_in_forward': False, 'rounding': 'nearest', 'fp16_mixed_quantize': False, 'quantize_change_ratio': 0.001}, 'different_groups': {}}, 'activation_quantization': {'shared_parameters': {'enabled': False, 'quantization_type': 'symmetric', 'range_calibration': 'dynamic', 'schedule_offset': 1000}, 'different_groups': {}}, 'sparse_pruning': {'shared_parameters': {'enabled': False, 'method': 'l1', 'schedule_offset': 1000}, 'different_groups': {}}, 'row_pruning': {'shared_parameters': {'enabled': False, 'method': 'l1', 'schedule_offset': 1000}, 'different_groups': {}}, 'head_pruning': {'shared_parameters': {'enabled': False, 'method': 'topk', 'schedule_offset': 1000}, 'different_groups': {}}, 'channel_pruning': {'shared_pa
+0: rameters': {'enabled': False, 'method': 'l1', 'schedule_offset': 1000}, 'different_groups': {}}, 'layer_reduction': {'enabled': False}}
+0: [2023-05-10 00:14:10,740] [INFO] [config.py:1011:print]   curriculum_enabled ........... False
+0: [2023-05-10 00:14:10,740] [INFO] [config.py:1011:print]   curriculum_params ............ False
+0: [2023-05-10 00:14:10,740] [INFO] [config.py:1011:print]   dataloader_drop_last ......... False
+0: [2023-05-10 00:14:10,740] [INFO] [config.py:1011:print]   disable_allgather ............ False
+0: [2023-05-10 00:14:10,740] [INFO] [config.py:1011:print]   dump_state ................... False
+0: [2023-05-10 00:14:10,740] [INFO] [config.py:1011:print]   dynamic_loss_scale_args ...... None
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   eigenvalue_enabled ........... False
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   eigenvalue_gas_boundary_resolution  1
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   eigenvalue_layer_name ........ bert.encoder.layer
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   eigenvalue_layer_num ......... 0
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   eigenvalue_max_iter .......... 100
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   eigenvalue_stability ......... 1e-06
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   eigenvalue_tol ............... 0.01
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   eigenvalue_verbose ........... False
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   elasticity_enabled ........... False
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   flops_profiler_config ........ {
+0:     "enabled": false, 
+0:     "profile_step": 1, 
+0:     "module_depth": -1, 
+0:     "top_modules": 1, 
+0:     "detailed": true, 
+0:     "output_file": null
+0: }
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   fp16_auto_cast ............... None
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   fp16_enabled ................. False
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   fp16_master_weights_and_gradients  False
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   global_rank .................. 0
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   gradient_accumulation_steps .. 1
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   gradient_clipping ............ 1.0
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   gradient_predivide_factor .... 1.0
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   initial_dynamic_scale ........ 1
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   load_universal_checkpoint .... False
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   loss_scale ................... 1.0
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   memory_breakdown ............. False
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   monitor_config ............... <deepspeed.monitor.config.DeepSpeedMonitorConfig object at 0x14b88d151ca0>
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   nebula_config ................ {
+0:     "enabled": false, 
+0:     "persistent_storage_path": null, 
+0:     "persistent_time_interval": 100, 
+0:     "num_of_version_in_retention": 2, 
+0:     "enable_nebula_load": true, 
+0:     "load_path": null
+0: }
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   optimizer_legacy_fusion ...... False
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   optimizer_name ............... None
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   optimizer_params ............. None
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   pipeline ..................... {'stages': 'auto', 'partition': 'best', 'seed_layers': False, 'activation_checkpoint_interval': 0}
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   pld_enabled .................. False
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   pld_params ................... False
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   prescale_gradients ........... False
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   scheduler_name ............... None
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   scheduler_params ............. None
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   sparse_attention ............. None
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   sparse_gradients_enabled ..... False
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   steps_per_print .............. 2000
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   train_batch_size ............. 256
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   train_micro_batch_size_per_gpu  32
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   use_node_local_storage ....... False
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   wall_clock_breakdown ......... False
+0: [2023-05-10 00:14:10,741] [INFO] [config.py:1011:print]   world_size ................... 8
+0: [2023-05-10 00:14:10,742] [INFO] [config.py:1011:print]   zero_allow_untested_optimizer  False
+0: [2023-05-10 00:14:10,742] [INFO] [config.py:1011:print]   zero_config .................. stage=0 contiguous_gradients=True reduce_scatter=True reduce_bucket_size=500000000 allgather_partitions=True allgather_bucket_size=500000000 overlap_comm=False load_from_fp32_weights=True elastic_checkpoint=False offload_param=None offload_optimizer=None sub_group_size=1000000000 cpu_offload_param=None cpu_offload_use_pin_memory=None cpu_offload=None prefetch_bucket_size=50000000 param_persistence_threshold=100000 model_persistence_threshold=9223372036854775807 max_live_parameters=1000000000 max_reuse_distance=1000000000 gather_16bit_weights_on_model_save=False stage3_gather_fp16_weights_on_model_save=False ignore_unused_parameters=True legacy_stage1=False round_robin_gradients=False
+0: [2023-05-10 00:14:10,742] [INFO] [config.py:1011:print]   zero_enabled ................. False
+0: [2023-05-10 00:14:10,742] [INFO] [config.py:1011:print]   zero_optimization_stage ...... 0
+0: [2023-05-10 00:14:10,742] [INFO] [config.py:996:print_user_config]   json = {
+0:     "train_micro_batch_size_per_gpu": 32, 
+0:     "train_batch_size": 256, 
+0:     "gradient_clipping": 1.0, 
+0:     "zero_optimization": {
+0:         "stage": 0
+0:     }, 
+0:     "bf16": {
+0:         "enabled": true
+0:     }, 
+0:     "steps_per_print": 2.000000e+03, 
+0:     "wall_clock_breakdown": false
+0: }
+0: Time to load utils op: 0.00042510032653808594 seconds
+0: [2023-05-10 00:14:10,742] [INFO] [engine.py:87:__init__] CONFIG: micro_batches=1 micro_batch_size=32
+0: [2023-05-10 00:14:10,754] [INFO] [engine.py:145:__init__] RANK=0 STAGE=0 LAYERS=12 [0, 12) STAGE_PARAMS=19703712 (19.704M) TOTAL_PARAMS=19703712 (19.704M) UNIQUE_PARAMS=19703712 (19.704M)
+0: [2023-05-10 00:14:10,765] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/mp_rank_00_model_states.pt...
+0: [2023-05-10 00:14:10,765] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/mp_rank_00_model_states.pt...
+0: [2023-05-10 00:14:10,765] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/mp_rank_00_model_states.pt...
+0: [2023-05-10 00:14:10,766] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/mp_rank_00_model_states.pt...
+0: [2023-05-10 00:14:10,766] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/mp_rank_00_model_states.pt...
+0: [2023-05-10 00:14:10,766] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/mp_rank_00_model_states.pt...
+0: [2023-05-10 00:14:10,766] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/mp_rank_00_model_states.pt...
+0: [2023-05-10 00:14:10,766] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/mp_rank_00_model_states.pt...
+0: [2023-05-10 00:14:10,855] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/mp_rank_00_model_states.pt.
+0: [2023-05-10 00:14:10,855] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/mp_rank_00_model_states.pt.
+0: [2023-05-10 00:14:10,855] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/mp_rank_00_model_states.pt.
+0: [2023-05-10 00:14:10,855] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/mp_rank_00_model_states.pt.
+0: [2023-05-10 00:14:10,855] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/mp_rank_00_model_states.pt.
+0: [2023-05-10 00:14:10,855] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/mp_rank_00_model_states.pt.
+0: [2023-05-10 00:14:10,855] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/mp_rank_00_model_states.pt.
+0: [2023-05-10 00:14:10,855] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/mp_rank_00_model_states.pt...
+0: [2023-05-10 00:14:10,855] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/mp_rank_00_model_states.pt...
+0: [2023-05-10 00:14:10,855] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/mp_rank_00_model_states.pt...
+0: [2023-05-10 00:14:10,855] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/mp_rank_00_model_states.pt...
+0: [2023-05-10 00:14:10,855] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/mp_rank_00_model_states.pt...
+0: [2023-05-10 00:14:10,855] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/mp_rank_00_model_states.pt...
+0: [2023-05-10 00:14:10,855] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/mp_rank_00_model_states.pt...
+0: [2023-05-10 00:14:10,855] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/mp_rank_00_model_states.pt.
+0: [2023-05-10 00:14:10,855] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/mp_rank_00_model_states.pt...
+0: [2023-05-10 00:14:10,856] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/mp_rank_00_model_states.pt.
+0: [2023-05-10 00:14:10,856] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/mp_rank_00_model_states.pt.
+0: [2023-05-10 00:14:10,856] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/mp_rank_00_model_states.pt.
+0: [2023-05-10 00:14:10,856] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/mp_rank_00_model_states.pt.
+0: [2023-05-10 00:14:10,856] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/mp_rank_00_model_states.pt.
+0: [2023-05-10 00:14:10,856] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/mp_rank_00_model_states.pt.
+0: [2023-05-10 00:14:10,856] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_01-model_00-model_states.pt...
+0: [2023-05-10 00:14:10,857] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/mp_rank_00_model_states.pt.
+0: [2023-05-10 00:14:10,857] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/mp_rank_00_model_states.pt.
+0: [2023-05-10 00:14:10,857] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_01-model_00-model_states.pt...
+0: [2023-05-10 00:14:10,857] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_01-model_00-model_states.pt...
+0: [2023-05-10 00:14:10,857] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_01-model_00-model_states.pt...
+0: [2023-05-10 00:14:10,857] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_01-model_00-model_states.pt...
+0: [2023-05-10 00:14:10,857] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_01-model_00-model_states.pt...
+0: [2023-05-10 00:14:10,857] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_01-model_00-model_states.pt...
+0: [2023-05-10 00:14:10,857] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_01-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,032] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_01-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,032] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_01-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,032] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_01-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,032] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_01-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,032] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_01-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,032] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_01-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,032] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_01-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,033] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_01-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,035] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_01-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,035] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_01-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,035] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_01-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,035] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_01-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,035] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_01-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,035] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_01-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,035] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_01-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,035] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_01-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,049] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_01-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,050] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_01-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,051] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_01-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,051] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_01-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,051] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_01-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,052] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_01-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,052] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_01-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,052] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_01-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,074] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_03-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,074] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_03-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,075] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_03-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,075] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_03-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,075] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_03-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,075] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_03-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,075] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_03-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,075] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_03-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,144] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_03-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,144] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_03-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,144] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_03-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,144] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_03-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,145] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_03-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,145] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_03-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,145] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_03-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,145] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_03-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,145] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_03-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,145] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_03-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,145] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_03-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,145] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_03-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,145] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_03-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,145] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_03-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,145] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_03-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,145] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_03-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,145] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_03-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,145] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_03-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,146] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_04-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,146] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_03-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,146] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_04-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,146] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_03-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,146] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_03-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,147] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_03-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,147] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_03-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,147] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_03-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,148] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_04-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,148] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_04-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,148] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_04-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,148] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_04-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,148] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_04-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,148] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_04-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,171] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_04-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,171] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_04-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,171] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_04-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,171] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_04-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,172] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_04-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,172] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_04-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,172] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_04-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,172] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_04-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,172] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_04-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,172] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_04-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,172] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_04-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,172] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_04-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,172] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_04-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,172] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_04-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,172] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_04-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,172] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_04-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,172] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_04-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,172] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_04-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,173] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_05-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,173] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_05-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,173] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_04-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,173] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_04-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,174] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_04-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,174] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_04-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,174] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_04-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,174] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_04-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,176] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_05-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,176] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_05-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,176] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_05-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,176] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_05-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,176] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_05-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,176] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_05-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,197] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_05-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,197] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_05-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,197] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_05-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,197] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_05-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,198] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_05-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,198] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_05-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,198] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_05-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,198] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_05-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,198] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_05-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,198] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_05-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,198] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_05-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,198] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_05-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,198] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_05-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,198] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_05-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,198] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_05-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,198] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_05-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,199] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_05-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,199] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_05-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,200] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_06-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,200] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_06-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,200] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_05-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,200] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_05-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,200] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_05-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,200] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_05-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,200] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_05-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,200] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_05-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,202] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_06-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,202] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_06-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,202] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_06-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,202] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_06-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,202] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_06-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,202] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_06-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,231] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_06-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,231] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_06-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,231] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_06-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,231] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_06-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,232] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_06-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,232] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_06-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,232] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_06-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,232] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_06-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,232] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_06-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,232] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_06-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,232] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_06-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,232] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_06-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,232] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_06-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,232] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_06-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,232] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_06-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,232] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_06-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,233] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_06-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,233] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_06-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,233] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_07-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,233] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_07-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,234] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_06-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,234] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_06-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,234] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_06-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,234] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_06-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,234] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_06-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,234] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_06-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,236] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_07-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,236] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_07-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,236] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_07-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,236] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_07-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,236] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_07-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,236] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_07-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,252] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_07-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,252] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_07-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,252] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_07-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,252] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_07-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,252] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_07-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,252] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_07-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,252] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_07-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,253] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_07-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,253] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_07-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,253] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_07-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,253] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_07-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,253] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_07-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,253] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_07-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,253] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_07-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,253] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_07-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,253] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_07-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,253] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_07-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,253] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_07-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,254] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_07-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,254] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_07-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,254] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_07-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,254] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_07-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,254] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_07-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,254] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_07-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,254] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_09-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,254] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_09-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,256] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_09-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,256] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_09-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,256] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_09-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,256] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_09-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,256] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_09-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,256] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_09-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,256] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_09-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,256] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_09-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,256] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_09-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,256] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_09-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,257] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_09-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,257] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_09-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,257] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_09-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,257] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_09-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,257] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_09-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,257] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_09-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,257] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_09-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,257] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_09-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,257] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_09-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,257] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_09-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,257] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_09-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,257] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_09-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,257] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_09-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,257] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/layer_09-model_00-model_states.pt...
+0: [2023-05-10 00:14:11,257] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_09-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,257] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_09-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,257] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_09-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,258] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_09-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,258] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_09-model_00-model_states.pt.
+0: [2023-05-10 00:14:11,258] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/layer_09-model_00-model_states.pt.
+0:  > overriding learning rate value to 0.0002
+0:  > overriding minimum learning rate value to 2e-05
+0:  > overriding warmup iterations value to 0
+0:  > overriding total number of iterations value to 1
+0:  > overriding decay style value to cosine
+0: [2023-05-10 00:14:11,260] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/bf16_zero_pp_rank_1_mp_rank_00_optim_states.pt...
+0: [2023-05-10 00:14:11,260] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/bf16_zero_pp_rank_0_mp_rank_00_optim_states.pt...
+0: [2023-05-10 00:14:11,260] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/bf16_zero_pp_rank_6_mp_rank_00_optim_states.pt...
+0: [2023-05-10 00:14:11,260] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/bf16_zero_pp_rank_2_mp_rank_00_optim_states.pt...
+0: [2023-05-10 00:14:11,260] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/bf16_zero_pp_rank_3_mp_rank_00_optim_states.pt...
+0: [2023-05-10 00:14:11,260] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/bf16_zero_pp_rank_4_mp_rank_00_optim_states.pt...
+0: [2023-05-10 00:14:11,260] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/bf16_zero_pp_rank_7_mp_rank_00_optim_states.pt...
+0: [2023-05-10 00:14:11,260] [INFO] [torch_checkpoint_engine.py:21:load] [Torch] Loading checkpoint from checkpoints_21m400m400m/global_step762/bf16_zero_pp_rank_5_mp_rank_00_optim_states.pt...
+0: [2023-05-10 00:14:11,363] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/bf16_zero_pp_rank_4_mp_rank_00_optim_states.pt.
+0: [2023-05-10 00:14:11,363] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 4
+0: [2023-05-10 00:14:11,365] [INFO] [engine.py:2784:_load_zero_checkpoint] loading 8 zero partition checkpoints for rank 4
+0: [2023-05-10 00:14:11,376] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/bf16_zero_pp_rank_3_mp_rank_00_optim_states.pt.
+0: [2023-05-10 00:14:11,376] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 3
+0: [2023-05-10 00:14:11,378] [INFO] [engine.py:2784:_load_zero_checkpoint] loading 8 zero partition checkpoints for rank 3
+0: [2023-05-10 00:14:11,387] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/bf16_zero_pp_rank_0_mp_rank_00_optim_states.pt.
+0: [2023-05-10 00:14:11,387] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 0
+0: [2023-05-10 00:14:11,390] [INFO] [engine.py:2784:_load_zero_checkpoint] loading 8 zero partition checkpoints for rank 0
+0: could not find arguments in the checkpoint ...
+0:  checkpoint version 3.0
+0: [2023-05-10 00:14:11,395] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/bf16_zero_pp_rank_5_mp_rank_00_optim_states.pt.
+0: [2023-05-10 00:14:11,395] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 5
+0: [2023-05-10 00:14:11,397] [INFO] [engine.py:2784:_load_zero_checkpoint] loading 8 zero partition checkpoints for rank 5
+0: [2023-05-10 00:14:11,398] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/bf16_zero_pp_rank_2_mp_rank_00_optim_states.pt.
+0: [2023-05-10 00:14:11,398] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 2
+0: [2023-05-10 00:14:11,400] [INFO] [engine.py:2784:_load_zero_checkpoint] loading 8 zero partition checkpoints for rank 2
+0: [2023-05-10 00:14:11,409] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/bf16_zero_pp_rank_1_mp_rank_00_optim_states.pt.
+0: [2023-05-10 00:14:11,409] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 1
+0: [2023-05-10 00:14:11,411] [INFO] [engine.py:2784:_load_zero_checkpoint] loading 8 zero partition checkpoints for rank 1
+0: [2023-05-10 00:14:11,444] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/bf16_zero_pp_rank_7_mp_rank_00_optim_states.pt.
+0: [2023-05-10 00:14:11,444] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 7
+0: [2023-05-10 00:14:11,445] [INFO] [engine.py:2784:_load_zero_checkpoint] loading 8 zero partition checkpoints for rank 7
+0: [2023-05-10 00:14:11,454] [INFO] [torch_checkpoint_engine.py:23:load] [Torch] Loaded checkpoint from checkpoints_21m400m400m/global_step762/bf16_zero_pp_rank_6_mp_rank_00_optim_states.pt.
+0: [2023-05-10 00:14:11,454] [INFO] [engine.py:2844:_get_all_zero_checkpoint_state_dicts] successfully read 8 ZeRO state_dicts for rank 6
+0: [2023-05-10 00:14:11,455] [INFO] [engine.py:2784:_load_zero_checkpoint] loading 8 zero partition checkpoints for rank 6
+0:   successfully loaded checkpoint from checkpoints_21m400m400m at iteration 0
+0: time (ms) | load-checkpoint: 701.61
+0: estimated model parameters: 0.019703712
+0: estimated model parameters without embeddings: 0.004626336
+0: [after model, optimizer, and learning rate scheduler are built] datetime: 2023-05-10 00:14:12 
+0: > building train, validation, and test datasets ...
+0:  > datasets target sizes (minimum size):
+0:     train:      1
+0:     validation: 25600
+0:     test:       25600
+0: > building train, validation, and test datasets for GPT ...
+0:  > building dataset index ...
+0:     reading sizes...
+0:     reading pointers...
+0:     reading document index...
+0:     creating numpy buffer of mmap...
+0:     creating memory view of numpy buffer...
+0:  > finished creating indexed dataset in 0.036932 seconds
+0:     number of documents: 835726
+0:  > dataset split:
+0:     train:
+0:      document indices in [0, 835726) total of 835726 documents
+0:  > loading doc-idx mapping from /scratch/project_462000119/data/c4_subsampled/gpt2tok_c4_en_400M_text_document_train_indexmap_1ns_2048sl_1234s_doc_idx.npy
+0:  > loading sample-idx mapping from /scratch/project_462000119/data/c4_subsampled/gpt2tok_c4_en_400M_text_document_train_indexmap_1ns_2048sl_1234s_sample_idx.npy
+0:  > loading shuffle-idx mapping from /scratch/project_462000119/data/c4_subsampled/gpt2tok_c4_en_400M_text_document_train_indexmap_1ns_2048sl_1234s_shuffle_idx.npy
+0:     loaded indexed file in 0.113 seconds
+0:     total number of samples: 195101
+0:     total number of epochs: 1
+0:  > building dataset index ...
+0:     reading sizes...
+0:     reading pointers...
+0:     reading document index...
+0:     creating numpy buffer of mmap...
+0:     creating memory view of numpy buffer...
+0:  > finished creating indexed dataset in 0.110965 seconds
+0:     number of documents: 364608
+0:  > dataset split:
+0:     validation:
+0:      document indices in [0, 364608) total of 364608 documents
+0:  > loading doc-idx mapping from /scratch/project_462000119/data/c4_validation/gpt2tok_c4validation_rerun_text_document_validation_indexmap_25600ns_2048sl_1234s_doc_idx.npy
+0:  > loading sample-idx mapping from /scratch/project_462000119/data/c4_validation/gpt2tok_c4validation_rerun_text_document_validation_indexmap_25600ns_2048sl_1234s_sample_idx.npy
+0:  > loading shuffle-idx mapping from /scratch/project_462000119/data/c4_validation/gpt2tok_c4validation_rerun_text_document_validation_indexmap_25600ns_2048sl_1234s_shuffle_idx.npy
+0:     loaded indexed file in 0.096 seconds
+0:     total number of samples: 84978
+0:     total number of epochs: 1
+0: > finished creating GPT datasets ...
+0: time (ms) | model-and-optimizer-setup: 9807.69 | train/valid/test-data-iterators-setup: 7326.47
+0: [after dataloaders are built] datetime: 2023-05-10 00:14:19 
+0: done with setup ...
+0: training ...
+0: [after training is done] datetime: 2023-05-10 00:14:19 
+0: -----------------------------------------------------------------------------------------------------------------
+0: validation loss at the end of training for val data | lm loss value: 6.096268E+00 | lm loss PPL: 4.441970E+02 | 
+0: -----------------------------------------------------------------------------------------------------------------
+END 3487337: Wed 10 May 2023 12:15:00 AM EEST

21m400m400m/global_step762/bf16_zero_pp_rank_0_mp_rank_00_optim_states.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ac95f09abe87dd9c68b9b4a9829080de26ac03ebe49e9fab6898e669e10204ca
+size 29560343

21m400m400m/global_step762/bf16_zero_pp_rank_1_mp_rank_00_optim_states.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:efb818b085f236a1a1f791c2f1a795db756a52a66ab08d0286bf5b8006b76794
+size 29560215

21m400m400m/global_step762/bf16_zero_pp_rank_2_mp_rank_00_optim_states.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6ef244aa36694f385dc28084076b72043fa3814eca2d07cf51b033500ef3989e
+size 29560343

21m400m400m/global_step762/bf16_zero_pp_rank_3_mp_rank_00_optim_states.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1317c9587f00bee62633072cbf5394ec9ee83838a437729fb54847887465bfee
+size 29560343

21m400m400m/global_step762/bf16_zero_pp_rank_4_mp_rank_00_optim_states.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6913ca4a0ae0e6991b668a9c7a92acc87ea502c5a5c096c07d8195a5cd06f69b
+size 29560279

21m400m400m/global_step762/bf16_zero_pp_rank_5_mp_rank_00_optim_states.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9ef2876867f1b52a3d6362517c9d58f2b70641956bff4a03cea9b7a09fabb83c
+size 29560343

21m400m400m/global_step762/bf16_zero_pp_rank_6_mp_rank_00_optim_states.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:47ed8dd698a7b82205d723acc34b2f90ff96e074b123b252f2db5bd902c18174
+size 29560471

21m400m400m/global_step762/bf16_zero_pp_rank_7_mp_rank_00_optim_states.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9f5f4b3628d526b6ca81f845847cc6c26fafc52052188540565a0478e03ebf0a
+size 29560727

21m400m400m/global_step762/layer_01-model_00-model_states.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:88174b136681638ce4da8cddad57423ca02d50894fedede8a82b5ed6b21df9a5
+size 30156035

21m400m400m/global_step762/layer_03-model_00-model_states.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:da290e4e9bdc0b0bf908d9233566700072a2485174cf204ca66121dc8e16afe1
+size 1854659

21m400m400m/global_step762/layer_04-model_00-model_states.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:248612036a8e9e67d8a9b5fd1ee266df4e0bc31afc9b7162f40554f7ebd5ff96
+size 1854659

21m400m400m/global_step762/layer_05-model_00-model_states.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c82a70fc53941272564fca38b712423a55b937d3b1395524d24a8e0b42cf2790
+size 1854659

21m400m400m/global_step762/layer_06-model_00-model_states.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fa99a8168c79f748cd8639466fd29bbb78f68c770fbf4b24763f6d72d2345b91
+size 1854659

21m400m400m/global_step762/layer_07-model_00-model_states.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3fb0d4bf06d9d68253b1b5556324c495446c93f1237ad79195f56ff4b1cf4ecf
+size 1854659

21m400m400m/global_step762/layer_09-model_00-model_states.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e2670512cabbec37a36b100140fc6e12d6f75b5c85108026ccf5971554ee687f
+size 2371

21m400m400m/global_step762/mp_rank_00_model_states.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a9a98d0d201d95a08d421020b41ee795e8f824d673cf1d88e81bdc3733007d47
+size 27827

21m400m400m/sbatch_21m400m400m.sh

@@ -0,0 +1,172 @@
+#!/bin/bash
+#SBATCH --exclude=nid007571,nid007112,nid006774,nid007502,nid007506,nid007507,nid005145,nid006692,nid007218,nid007123,nid006124,nid006123,nid007496,nid007237,nid006852,nid007206,nid006947,nid007212,nid006977,nid007222,nid005444,nid007219,nid007493,nid007221,nid005300,nid005619,nid006118,nid005203,nid006113,nid006481,nid007077,nid005208,nid005207,nid005879,nid005901
+#SBATCH --nodes=1
+#SBATCH --ntasks-per-node=1
+#SBATCH --cpus-per-task=32
+#SBATCH --mem=256G
+#SBATCH -p standard-g
+#SBATCH -t 2-0:00:00
+#SBATCH --gpus-per-node=mi250:8
+#SBATCH --exclusive=user
+#SBATCH --hint=nomultithread
+#SBATCH --account=project_462000119
+#SBATCH -o logs/%j.out
+#SBATCH -e logs/%j.err
+
+VARIANT=21m400m400m
+
+# if run without sbatch, invoke here
+if [ -z $SLURM_JOB_ID ]; then
+    mkdir -p logs
+    sbatch "$0"
+    exit
+fi
+
+set -euo pipefail
+
+# symlink logs/latest.out and logs/latest.err
+ln -f -s $SLURM_JOB_ID.out logs/latest.out
+ln -f -s $SLURM_JOB_ID.err logs/latest.err
+
+KILL_SWITCH_PATH=kill-switch-$VARIANT
+CHECKPOINT_PATH=checkpoints_$VARIANT
+TENSORBOARD_PATH=tensorboard_$VARIANT
+mkdir -p $CHECKPOINT_PATH
+mkdir -p $TENSORBOARD_PATH
+
+# Data
+VOCAB_FILE="gpt2/vocab.json"
+MERGE_FILE="gpt2/merges.txt"
+#DATA_PATH="/scratch/project_462000119/data/pile/megatron_data/meg-gpt2_pile_text_document"
+TRAIN_DATA_PATH=train400m.txt
+# "train: 1.0 0:1 /scratch/project_462000119/data/c4_subsampled/gpt2tok_c4_en_1B5_text_document"
+VALID_DATA_PATH=val.txt
+# "validation: 1.0 0:1 /scratch/project_462000119/data/c4_validation/gpt2tok_c4validation_rerun_text_document"
+
+
+PP_SIZE=1
+TP_SIZE=1
+
+MICRO_BATCH_SIZE=32
+GRADIENT_ACCUMULATION_STEPS=1
+WORLD_SIZE=$((SLURM_GPUS_ON_NODE*SLURM_JOB_NUM_NODES))
+GLOBAL_BATCH_SIZE=$((MICRO_BATCH_SIZE*WORLD_SIZE*GRADIENT_ACCUMULATION_STEPS))
+
+# Model parameters
+source model_params.sh
+MODEL_PARAM=("${PARAM_21M[@]}")
+NHIDDEN=${MODEL_PARAM[0]}
+FFN_HIDDEN_SIZE=${MODEL_PARAM[1]}
+KV_SIZE=${MODEL_PARAM[2]}
+NHEADS=${MODEL_PARAM[3]}
+NLAYERS=${MODEL_PARAM[4]}
+SEQ_LEN=2048
+
+echo "Model parameters: d_model $NHIDDEN ffw_size $FFN_HIDDEN_SIZE kv_size $KV_SIZE n_heads $NHEADS n_layers $NLAYERS"
+
+SAVE_INTERVAL=10000
+
+# Tokens: 100 000 000
+# -> Samples: 48828.125
+#TRAIN_SAMPLES=48_828
+# Tokens: 400M
+# 195312.5
+TRAIN_SAMPLES=195_313
+
+
+OPTIMIZER_ARGS=" \
+    --optimizer adam \
+    --adam-beta1 0.9 \
+    --adam-beta2 0.999 \
+    --adam-eps 1e-8 \
+    --lr 2e-4 \
+    --min-lr 2e-5 \
+    --lr-decay-style cosine \
+    --lr-decay-samples $TRAIN_SAMPLES \
+    --lr-warmup-samples 1953 \
+    --clip-grad 1.0 \
+    --weight-decay 1e-1 \
+    "
+
+GPT_ARGS=" \
+    --num-layers $NLAYERS \
+    --hidden-size $NHIDDEN \
+    --num-attention-heads $NHEADS \
+    --kv-channels $KV_SIZE \
+    --ffn-hidden-size $FFN_HIDDEN_SIZE \
+    --seq-length $SEQ_LEN \
+    --max-position-embeddings $SEQ_LEN \
+    --micro-batch-size $MICRO_BATCH_SIZE \
+    --global-batch-size $GLOBAL_BATCH_SIZE \
+    --train-samples $TRAIN_SAMPLES \
+    --vocab-file $VOCAB_FILE \
+    --merge-file $MERGE_FILE \
+    --loss-scale 12 \
+    --clip-grad 1.0 \
+    --kill-switch-path $KILL_SWITCH_PATH \
+    --bf16 \
+    --checkpoint-activations \
+    $OPTIMIZER_ARGS \
+    "
+
+OUTPUT_ARGS=" \
+    --log-interval 10 \
+    --save-interval $SAVE_INTERVAL \
+    --eval-interval 1000 \
+    --eval-iters 1 \
+    --tensorboard-dir $TENSORBOARD_PATH \
+    --tensorboard-queue-size 5 \
+    --log-timers-to-tensorboard \
+    --log-batch-size-to-tensorboard \
+    --log-validation-ppl-to-tensorboard \
+    "
+
+ZERO_STAGE=0
+
+mkdir -p ds_configs
+DS_CONFIG_PATH="ds_configs/$SLURM_JOB_ID.json"
+
+cat <<EOF > $DS_CONFIG_PATH
+{
+    "train_micro_batch_size_per_gpu": $MICRO_BATCH_SIZE,
+    "train_batch_size": $GLOBAL_BATCH_SIZE,
+    "gradient_clipping": 1.0,
+    "zero_optimization": {
+        "stage": $ZERO_STAGE
+    },
+    "bf16": {
+        "enabled": true
+    },
+    "steps_per_print": 2000,
+    "wall_clock_breakdown": false
+}
+EOF
+
+DEEPSPEED_ARGS=" \
+    --deepspeed \
+    --deepspeed_config $DS_CONFIG_PATH \
+    --zero-stage $ZERO_STAGE \
+    "
+
+CMD=" \
+    Megatron-DeepSpeed/pretrain_gpt.py \
+    --tensor-model-parallel-size $TP_SIZE \
+    --pipeline-model-parallel-size $PP_SIZE \
+    $GPT_ARGS \
+    $OUTPUT_ARGS \
+    --save $CHECKPOINT_PATH \
+    --load $CHECKPOINT_PATH \
+    --train-weighted-split-paths-path $TRAIN_DATA_PATH \
+    --valid-weighted-split-paths-path $VALID_DATA_PATH \
+    --data-impl mmap \
+     $DEEPSPEED_ARGS \
+    "
+
+echo $CMD
+
+echo "START $SLURM_JOBID: $(date)"
+
+# bash launch_srun.sh $CMD
+srun --label launch.sh $CMD
+
+echo "END $SLURM_JOBID: $(date)"

21m400m400m/sbatch_21m400m400mval.sh

@@ -0,0 +1,167 @@
+#!/bin/bash
+#SBATCH --exclude=nid007571,nid007112,nid006774,nid007502,nid007506,nid007507,nid005145,nid006692,nid007218,nid007123,nid006124,nid006123,nid007496,nid007237,nid006852,nid007206,nid006947,nid007212,nid006977,nid007222,nid005444,nid007219,nid007493,nid007221,nid005300,nid005619,nid006118,nid005203,nid006113,nid006481,nid007077,nid005208,nid005207,nid005879,nid005901
+#SBATCH --nodes=1
+#SBATCH --ntasks-per-node=1
+#SBATCH --cpus-per-task=32
+#SBATCH --mem=256G
+#SBATCH -p small-g
+#SBATCH -t 12:00:00
+#SBATCH --gpus-per-node=mi250:8
+#SBATCH --exclusive=user
+#SBATCH --hint=nomultithread
+#SBATCH --account=project_462000119
+#SBATCH -o logs/%j.out
+#SBATCH -e logs/%j.err
+
+VARIANT=21m400m400mval
+VARIANT_CKPT=21m400m400m
+
+# if run without sbatch, invoke here
+if [ -z $SLURM_JOB_ID ]; then
+    mkdir -p logs
+    sbatch "$0"
+    exit
+fi
+
+set -euo pipefail
+
+# symlink logs/latest.out and logs/latest.err
+ln -f -s $SLURM_JOB_ID.out logs/latest.out
+ln -f -s $SLURM_JOB_ID.err logs/latest.err
+
+KILL_SWITCH_PATH=kill-switch-$VARIANT
+CHECKPOINT_PATH=checkpoints_$VARIANT_CKPT
+TENSORBOARD_PATH=tensorboard_$VARIANT
+
+# Data
+VOCAB_FILE="gpt2/vocab.json"
+MERGE_FILE="gpt2/merges.txt"
+#DATA_PATH="/scratch/project_462000119/data/pile/megatron_data/meg-gpt2_pile_text_document"
+TRAIN_DATA_PATH=train400m.txt
+# "train: 1.0 0:1 /scratch/project_462000119/data/c4_subsampled/gpt2tok_c4_en_4B8_text_document"
+VALID_DATA_PATH=val.txt
+# "validation: 1.0 0:1 /scratch/project_462000119/data/c4_validation/gpt2tok_c4validation_rerun_text_document"
+
+PP_SIZE=1
+TP_SIZE=1
+
+MICRO_BATCH_SIZE=32
+GRADIENT_ACCUMULATION_STEPS=1
+WORLD_SIZE=$((SLURM_GPUS_ON_NODE*SLURM_JOB_NUM_NODES))
+GLOBAL_BATCH_SIZE=$((MICRO_BATCH_SIZE*WORLD_SIZE*GRADIENT_ACCUMULATION_STEPS))
+
+# Model parameters
+source model_params.sh
+MODEL_PARAM=("${PARAM_20M[@]}")
+NHIDDEN=${MODEL_PARAM[0]}
+FFN_HIDDEN_SIZE=${MODEL_PARAM[1]}
+KV_SIZE=${MODEL_PARAM[2]}
+NHEADS=${MODEL_PARAM[3]}
+NLAYERS=${MODEL_PARAM[4]}
+SEQ_LEN=2048
+
+echo "Model parameters: d_model $NHIDDEN ffw_size $FFN_HIDDEN_SIZE kv_size $KV_SIZE n_heads $NHEADS n_layers $NLAYERS"
+
+SAVE_INTERVAL=1000
+
+# Tokens: 31633480000
+# -> Samples: 15446035
+TRAIN_SAMPLES=1
+
+OPTIMIZER_ARGS=" \
+    --optimizer adam \
+    --adam-beta1 0.9 \
+    --adam-beta2 0.999 \
+    --adam-eps 1e-8 \
+    --lr 2e-4 \
+    --min-lr 2e-5 \
+    --lr-decay-style cosine \
+    --lr-decay-samples $TRAIN_SAMPLES \
+    --lr-warmup-samples 0 \
+    --clip-grad 1.0 \
+    --weight-decay 1e-1 \
+    --override-lr-scheduler \
+    --no-load-optim \
+    --reset-progress \
+    "
+
+GPT_ARGS=" \
+    --num-layers $NLAYERS \
+    --hidden-size $NHIDDEN \
+    --num-attention-heads $NHEADS \
+    --kv-channels $KV_SIZE \
+    --ffn-hidden-size $FFN_HIDDEN_SIZE \
+    --seq-length $SEQ_LEN \
+    --max-position-embeddings $SEQ_LEN \
+    --micro-batch-size $MICRO_BATCH_SIZE \
+    --global-batch-size $GLOBAL_BATCH_SIZE \
+    --train-samples $TRAIN_SAMPLES \
+    --vocab-file $VOCAB_FILE \
+    --merge-file $MERGE_FILE \
+    --clip-grad 1.0 \
+    --kill-switch-path $KILL_SWITCH_PATH \
+    --bf16 \
+    $OPTIMIZER_ARGS \
+    "
+
+OUTPUT_ARGS=" \
+    --log-interval 10 \
+    --save-interval $SAVE_INTERVAL \
+    --eval-interval 1 \
+    --eval-iters 100 \
+    --tensorboard-dir $TENSORBOARD_PATH \
+    --tensorboard-queue-size 5 \
+    --log-timers-to-tensorboard \
+    --log-batch-size-to-tensorboard \
+    --log-validation-ppl-to-tensorboard \
+    "
+
+ZERO_STAGE=0
+
+mkdir -p ds_configs
+DS_CONFIG_PATH="ds_configs/$SLURM_JOB_ID.json"
+
+cat <<EOF > $DS_CONFIG_PATH
+{
+    "train_micro_batch_size_per_gpu": $MICRO_BATCH_SIZE,
+    "train_batch_size": $GLOBAL_BATCH_SIZE,
+    "gradient_clipping": 1.0,
+    "zero_optimization": {
+        "stage": $ZERO_STAGE
+    },
+    "bf16": {
+        "enabled": true
+    },
+    "steps_per_print": 2000,
+    "wall_clock_breakdown": false
+}
+EOF
+
+DEEPSPEED_ARGS=" \
+    --deepspeed \
+    --deepspeed_config $DS_CONFIG_PATH \
+    --zero-stage $ZERO_STAGE \
+    "
+
+CMD=" \
+    Megatron-DeepSpeed/pretrain_gpt.py \
+    --tensor-model-parallel-size $TP_SIZE \
+    --pipeline-model-parallel-size $PP_SIZE \
+    $GPT_ARGS \
+    $OUTPUT_ARGS \
+    --save $CHECKPOINT_PATH \
+    --load $CHECKPOINT_PATH \
+    --train-weighted-split-paths-path $TRAIN_DATA_PATH \
+    --valid-weighted-split-paths-path $VALID_DATA_PATH \
+    --data-impl mmap \
+     $DEEPSPEED_ARGS \
+    "
+
+echo $CMD
+
+echo "START $SLURM_JOBID: $(date)"
+
+# bash launch_srun.sh $CMD
+srun --label launch.sh $CMD
+
+echo "END $SLURM_JOBID: $(date)"

21m400m400m/tensorboard_21m400m400m/events.out.tfevents.1683665937.nid005223.63334.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e21f3a911a130ca44d1743a160abe95a79e77640e38e8542149a3c89501d6f03
+size 1369058

21m400m400m/tensorboard_21m400m400mval/events.out.tfevents.1683666595.nid007269.50912.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:65257e05e3a25f4bf5fa93218d0fc6e10e8040d0f4501e532d87436fb186a6e6
+size 40

21m400m400m/tensorboard_21m400m400mval/events.out.tfevents.1683666830.nid007269.54799.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6e683f0205510ce039d2a2d0816282b94c3cdfb5b71fdac91e5280090b1a264b
+size 980

220m200b1b5/sbatch_220m200b1b5.sh

@@ -0,0 +1,166 @@
+#!/bin/bash
+#SBATCH --nodes=8
+#SBATCH --ntasks-per-node=1
+#SBATCH --cpus-per-task=32
+#SBATCH --mem=256G
+#SBATCH -p standard-g
+#SBATCH -t 48:00:00
+#SBATCH --gpus-per-node=mi250:8
+#SBATCH --exclusive=user
+#SBATCH --hint=nomultithread
+#SBATCH --account=project_462000119
+#SBATCH -o logs/%j.out
+#SBATCH -e logs/%j.err
+
+VARIANT=220m200b1b5
+
+# if run without sbatch, invoke here
+if [ -z $SLURM_JOB_ID ]; then
+    mkdir -p logs
+    sbatch "$0"
+    exit
+fi
+
+set -euo pipefail
+
+# symlink logs/latest.out and logs/latest.err
+ln -f -s $SLURM_JOB_ID.out logs/latest.out
+ln -f -s $SLURM_JOB_ID.err logs/latest.err
+
+KILL_SWITCH_PATH=kill-switch-$VARIANT
+CHECKPOINT_PATH=checkpoints_$VARIANT
+TENSORBOARD_PATH=tensorboard_$VARIANT
+# Start from scratch
+# rm -rf "$CHECKPOINT_PATH" "$TENSORBOARD_PATH"
+
+# Data
+VOCAB_FILE="gpt2/vocab.json"
+MERGE_FILE="gpt2/merges.txt"
+# DATA_PATH="/scratch/project_462000119/data/pile/megatron_data/meg-gpt2_pile_text_document"
+TRAIN_DATA_PATH=train1b5.txt
+# "train: 1.0 0:1 /scratch/project_462000119/data/c4_subsampled/gpt2tok_c4_en_1B5_text_document"
+VALID_DATA_PATH=val.txt
+# "validation: 1.0 0:1 /scratch/project_462000119/data/c4_validation/gpt2tok_c4validation_rerun_text_document"
+
+PP_SIZE=1
+TP_SIZE=1
+
+MICRO_BATCH_SIZE=4
+GRADIENT_ACCUMULATION_STEPS=1
+WORLD_SIZE=$((SLURM_GPUS_ON_NODE*SLURM_JOB_NUM_NODES))
+GLOBAL_BATCH_SIZE=$((MICRO_BATCH_SIZE*WORLD_SIZE*GRADIENT_ACCUMULATION_STEPS))
+
+# Model parameters
+source model_params.sh
+MODEL_PARAM=("${PARAM_217M[@]}")
+NHIDDEN=${MODEL_PARAM[0]}
+FFN_HIDDEN_SIZE=${MODEL_PARAM[1]}
+KV_SIZE=${MODEL_PARAM[2]}
+NHEADS=${MODEL_PARAM[3]}
+NLAYERS=${MODEL_PARAM[4]}
+SEQ_LEN=2048
+
+echo "Model parameters: d_model $NHIDDEN ffw_size $FFN_HIDDEN_SIZE kv_size $KV_SIZE n_heads $NHEADS n_layers $NLAYERS"
+
+SAVE_INTERVAL=20000
+
+# Tokens: 7510000000
+# -> Samples: 3_666_992
+TRAIN_SAMPLES=3_666_992
+# Tokens: 200e9
+TRAIN_SAMPLES=97_656_250
+
+OPTIMIZER_ARGS=" \
+    --optimizer adam \
+    --adam-beta1 0.9 \
+    --adam-beta2 0.999 \
+    --adam-eps 1e-8 \
+    --lr 2e-4 \
+    --min-lr 2e-5 \
+    --lr-decay-style cosine \
+    --lr-decay-samples $TRAIN_SAMPLES \
+    --lr-warmup-samples 976_563 \
+    --clip-grad 1.0 \
+    --weight-decay 1e-1 \
+    "
+
+GPT_ARGS=" \
+    --num-layers $NLAYERS \
+    --hidden-size $NHIDDEN \
+    --num-attention-heads $NHEADS \
+    --kv-channels $KV_SIZE \
+    --ffn-hidden-size $FFN_HIDDEN_SIZE \
+    --seq-length $SEQ_LEN \
+    --max-position-embeddings $SEQ_LEN \
+    --micro-batch-size $MICRO_BATCH_SIZE \
+    --global-batch-size $GLOBAL_BATCH_SIZE \
+    --train-samples $TRAIN_SAMPLES \
+    --vocab-file $VOCAB_FILE \
+    --merge-file $MERGE_FILE \
+    --clip-grad 1.0 \
+    --kill-switch-path $KILL_SWITCH_PATH \
+    --bf16 \
+    $OPTIMIZER_ARGS \
+    "
+
+OUTPUT_ARGS=" \
+    --log-interval 500 \
+    --save-interval $SAVE_INTERVAL \
+    --eval-interval 50000 \
+    --eval-iters 1 \
+    --tensorboard-dir $TENSORBOARD_PATH \
+    --tensorboard-queue-size 5 \
+    --log-timers-to-tensorboard \
+    --log-batch-size-to-tensorboard \
+    --log-validation-ppl-to-tensorboard \
+    "
+
+ZERO_STAGE=0
+
+mkdir -p ds_configs
+DS_CONFIG_PATH="ds_configs/$SLURM_JOB_ID.json"
+
+cat <<EOF > $DS_CONFIG_PATH
+{
+    "train_micro_batch_size_per_gpu": $MICRO_BATCH_SIZE,
+    "train_batch_size": $GLOBAL_BATCH_SIZE,
+    "gradient_clipping": 1.0,
+    "zero_optimization": {
+        "stage": $ZERO_STAGE
+    },
+    "bf16": {
+        "enabled": true
+    },
+    "steps_per_print": 2000,
+    "wall_clock_breakdown": false
+}
+EOF
+
+DEEPSPEED_ARGS=" \
+    --deepspeed \
+    --deepspeed_config $DS_CONFIG_PATH \
+    --zero-stage $ZERO_STAGE \
+    "
+
+CMD=" \
+    Megatron-DeepSpeed2/pretrain_gpt.py \
+    --tensor-model-parallel-size $TP_SIZE \
+    --pipeline-model-parallel-size $PP_SIZE \
+    $GPT_ARGS \
+    $OUTPUT_ARGS \
+    --save $CHECKPOINT_PATH \
+    --load $CHECKPOINT_PATH \
+    --train-weighted-split-paths-path $TRAIN_DATA_PATH \
+    --valid-weighted-split-paths-path $VALID_DATA_PATH \
+    --data-impl mmap \
+     $DEEPSPEED_ARGS \
+    "
+
+echo $CMD
+
+echo "START $SLURM_JOBID: $(date)"
+
+# bash launch_srun.sh $CMD
+srun --label launch.sh $CMD
+
+echo "END $SLURM_JOBID: $(date)"

220m200b1b5/sbatch_220m200b1b5val.sh

@@ -0,0 +1,167 @@
+#!/bin/bash
+#SBATCH --nodes=8
+#SBATCH --ntasks-per-node=1
+#SBATCH --cpus-per-task=32
+#SBATCH --mem=256G
+#SBATCH -p standard-g
+#SBATCH -t 48:00:00
+#SBATCH --gpus-per-node=mi250:8
+#SBATCH --exclusive=user
+#SBATCH --hint=nomultithread
+#SBATCH --account=project_462000119
+#SBATCH -o logs/%j.out
+#SBATCH -e logs/%j.err
+
+VARIANT=220m200b1b5val
+VARIANT_CKPT=220m200b1b5
+
+# if run without sbatch, invoke here
+if [ -z $SLURM_JOB_ID ]; then
+    mkdir -p logs
+    sbatch "$0"
+    exit
+fi
+
+set -euo pipefail
+
+# symlink logs/latest.out and logs/latest.err
+ln -f -s $SLURM_JOB_ID.out logs/latest.out
+ln -f -s $SLURM_JOB_ID.err logs/latest.err
+
+KILL_SWITCH_PATH=kill-switch-$VARIANT
+CHECKPOINT_PATH=checkpoints_$VARIANT_CKPT
+TENSORBOARD_PATH=tensorboard_$VARIANT
+
+# Data
+VOCAB_FILE="gpt2/vocab.json"
+MERGE_FILE="gpt2/merges.txt"
+# DATA_PATH="/scratch/project_462000119/data/pile/megatron_data/meg-gpt2_pile_text_document"
+TRAIN_DATA_PATH=train1b5.txt
+# "train: 1.0 0:1 /scratch/project_462000119/data/c4_subsampled/gpt2tok_c4_en_7B5_text_document"
+VALID_DATA_PATH=val.txt
+# "validation: 1.0 0:1 /scratch/project_462000119/data/c4_validation/gpt2tok_c4validation_rerun_text_document"
+
+PP_SIZE=1
+TP_SIZE=1
+
+MICRO_BATCH_SIZE=4
+GRADIENT_ACCUMULATION_STEPS=1
+WORLD_SIZE=$((SLURM_GPUS_ON_NODE*SLURM_JOB_NUM_NODES))
+GLOBAL_BATCH_SIZE=$((MICRO_BATCH_SIZE*WORLD_SIZE*GRADIENT_ACCUMULATION_STEPS))
+
+# Model parameters
+source model_params.sh
+MODEL_PARAM=("${PARAM_217M[@]}")
+NHIDDEN=${MODEL_PARAM[0]}
+FFN_HIDDEN_SIZE=${MODEL_PARAM[1]}
+KV_SIZE=${MODEL_PARAM[2]}
+NHEADS=${MODEL_PARAM[3]}
+NLAYERS=${MODEL_PARAM[4]}
+SEQ_LEN=2048
+
+echo "Model parameters: d_model $NHIDDEN ffw_size $FFN_HIDDEN_SIZE kv_size $KV_SIZE n_heads $NHEADS n_layers $NLAYERS"
+
+SAVE_INTERVAL=1000
+
+# Tokens: 7510000000
+# -> Samples: 3_666_992
+TRAIN_SAMPLES=1
+
+OPTIMIZER_ARGS=" \
+    --optimizer adam \
+    --adam-beta1 0.9 \
+    --adam-beta2 0.999 \
+    --adam-eps 1e-8 \
+    --lr 2e-4 \
+    --min-lr 2e-5 \
+    --lr-decay-style cosine \
+    --lr-decay-samples $TRAIN_SAMPLES \
+    --lr-warmup-samples 0 \
+    --clip-grad 1.0 \
+    --weight-decay 1e-1 \
+    --no-load-optim \
+    --reset-progress \
+    --override-lr-scheduler \
+    "
+
+GPT_ARGS=" \
+    --num-layers $NLAYERS \
+    --hidden-size $NHIDDEN \
+    --num-attention-heads $NHEADS \
+    --kv-channels $KV_SIZE \
+    --ffn-hidden-size $FFN_HIDDEN_SIZE \
+    --seq-length $SEQ_LEN \
+    --max-position-embeddings $SEQ_LEN \
+    --micro-batch-size $MICRO_BATCH_SIZE \
+    --global-batch-size $GLOBAL_BATCH_SIZE \
+    --train-samples $TRAIN_SAMPLES \
+    --vocab-file $VOCAB_FILE \
+    --merge-file $MERGE_FILE \
+    --clip-grad 1.0 \
+    --kill-switch-path $KILL_SWITCH_PATH \
+    --bf16 \
+    $OPTIMIZER_ARGS \
+    "
+
+OUTPUT_ARGS=" \
+    --log-interval 10 \
+    --save-interval $SAVE_INTERVAL \
+    --eval-interval 1 \
+    --eval-only true \
+    --eval-iters 100 \
+    --tensorboard-dir $TENSORBOARD_PATH \
+    --tensorboard-queue-size 5 \
+    --log-timers-to-tensorboard \
+    --log-batch-size-to-tensorboard \
+    --log-validation-ppl-to-tensorboard \
+    "
+
+ZERO_STAGE=0
+
+mkdir -p ds_configs
+DS_CONFIG_PATH="ds_configs/$SLURM_JOB_ID.json"
+
+cat <<EOF > $DS_CONFIG_PATH
+{
+    "train_micro_batch_size_per_gpu": $MICRO_BATCH_SIZE,
+    "train_batch_size": $GLOBAL_BATCH_SIZE,
+    "gradient_clipping": 1.0,
+    "zero_optimization": {
+        "stage": $ZERO_STAGE
+    },
+    "bf16": {
+        "enabled": true
+    },
+    "steps_per_print": 2000,
+    "wall_clock_breakdown": false
+}
+EOF
+
+DEEPSPEED_ARGS=" \
+    --deepspeed \
+    --deepspeed_config $DS_CONFIG_PATH \
+    --zero-stage $ZERO_STAGE \
+    "
+
+CMD=" \
+    Megatron-DeepSpeed/pretrain_gpt.py \
+    --tensor-model-parallel-size $TP_SIZE \
+    --pipeline-model-parallel-size $PP_SIZE \
+    $GPT_ARGS \
+    $OUTPUT_ARGS \
+    --save $CHECKPOINT_PATH \
+    --load $CHECKPOINT_PATH \
+    --train-weighted-split-paths-path $TRAIN_DATA_PATH \
+    --valid-weighted-split-paths-path $VALID_DATA_PATH \
+    --data-impl mmap \
+     $DEEPSPEED_ARGS \
+    "
+
+echo $CMD
+
+echo "START $SLURM_JOBID: $(date)"
+
+# bash launch_srun.sh $CMD
+srun --label launch.sh $CMD
+
+echo "END $SLURM_JOBID: $(date)"

220m200b1b5/tensorboard_220m200b1b5/events.out.tfevents.1679051664.nid006529.96495.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4d8a384f1edb99245e32c8e09b0b98d0e785f7fcc795b02ae7979e700b7bf876
+size 40

220m200b1b5/tensorboard_220m200b1b5/events.out.tfevents.1679051664.nid006860.1508.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b5e7546168f06f9f3997d509636c095c97ad934478f199f0fb05746159f099a3
+size 40

220m200b1b5/tensorboard_220m200b1b5/events.out.tfevents.1679054214.nid005116.13183.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:86485eaeafb6eadfdec07df2066d12246f64fd1e41a62d7889ad18b31b08e77d
+size 40

220m200b1b5/tensorboard_220m200b1b5/events.out.tfevents.1680793013.nid006063.122512.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:54a286529cce0b3c873c9a9f87735d562748c34cdd700798b63d72f4c6cf4b04
+size 40

220m200b1b5/tensorboard_220m200b1b5/events.out.tfevents.1680793015.nid006273.117983.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:14a468073411f7d913b12b12958d64739bbd2a3c29f1c94b9ecfc8d325ed5da8
+size 40

220m200b1b5/tensorboard_220m200b1b5/events.out.tfevents.1680793016.nid005651.127873.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:207df892df43e9394cb11e2ff1b4f7f95d0039b25b74c6a900e0a8bd5208c8d7
+size 40

220m200b1b5/tensorboard_220m200b1b5/events.out.tfevents.1680793016.nid006265.117563.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fe26d220f3723b40d1eecd1dff70d045ca64c392ecc445005c44ea1a44a7c8fd
+size 40

220m200b1b5/tensorboard_220m200b1b5/events.out.tfevents.1680793017.nid006567.56933.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:107639576d96fa795148cb8550a24f8fb4843cb1fea76884db17ed6a8830cb0d
+size 40

220m200b1b5/tensorboard_220m200b1b5/events.out.tfevents.1680793017.nid006575.55528.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5b7b3238afa138c1766b78154d5126f367ee9ee615cbceca0031b11ef351dde6
+size 40

220m200b1b5/tensorboard_220m200b1b5/events.out.tfevents.1680793022.nid005643.128637.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:eb81c8fea8287a723a191dcd2cac25f58dae45ecb292239565daeff7b5744b59
+size 40

220m200b1b5/tensorboard_220m200b1b5/events.out.tfevents.1680793023.nid005499.130838.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4c37ee75762d5906a51e76ad3873f3c60ca442e17373bfaa1ab28f2a8bf994cb
+size 40

220m200b1b5/tensorboard_220m200b1b5/events.out.tfevents.1680793030.nid006090.118840.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6d4c4baad370c93e5d77d108957721ff2d317a621ef1af7d7ff68613b258bc9d
+size 40

220m200b1b5/tensorboard_220m200b1b5/events.out.tfevents.1680793032.nid006082.119082.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2dc49565466861ca44cbcc7856fbd1e612b5ef85984b213f5bcd98f029edfac3
+size 40

220m200b1b5/tensorboard_220m200b1b5val/events.out.tfevents.1679048250.nid005617.100428.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f54193a536aeffe88bfa5a335d40b17950d0849e95ee52fed089d78d34b17375
+size 980

2b812b4b/tensorboard_2b812b4bval/events.out.tfevents.1683534410.nid005943.115511.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3f30d1b2ad94b178af57f6d4acf4195d63335c5830d80ccca348d5f14396be1b
+size 980

2b816b4b/tensorboard_2b816b4bval/events.out.tfevents.1683561719.nid006565.21977.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d1cc546baa926fa8573d48c6b8a9b97e0852cffcc2f6a5b9d670e04b65412704
+size 980

4b248b12b/global_step45776/bf16_zero_pp_rank_0_mp_rank_00_optim_states.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0e5a466022447ef149b3b3c49260ed33208121c2bbd10b475858ef541a553b07
+size 199058647

4b248b12b/global_step45776/bf16_zero_pp_rank_0_mp_rank_01_optim_states.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e9377c402647d084fe9ee9c4767817b75eaed42cf13d5b0723e40080a55b5faf
+size 199058647

4b248b12b/global_step45776/bf16_zero_pp_rank_100_mp_rank_00_optim_states.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a67f8bdad178e3d68ab5266bd869a9c5d7a3d31bb2fb0c50fbf134ef8a713d01
+size 199058733

4b248b12b/global_step45776/bf16_zero_pp_rank_100_mp_rank_01_optim_states.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:270ebd675f83f6d8795726eb380cf4d49936061eea440afea1a7a8e6a8ff3ea8
+size 199058733

4b248b12b/global_step45776/bf16_zero_pp_rank_101_mp_rank_00_optim_states.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3b535b3009b9c39e0f2cae44eae00ebaf3a7dbfa970ed32c2c65878172e6c3e1
+size 199058797