CPU memory update for SMMP GPT2

aws · May 26, 2022 · e3a4e03 · e3a4e03
2 parents d403a33 + b704aaa
commit e3a4e03
Show file tree

Hide file tree

Showing 31 changed files with 783 additions and 4,768 deletions.
diff --git a/aws_sagemaker_studio/frameworks/keras_pipe_mode_horovod/generate_cifar10_tfrecords.py b/aws_sagemaker_studio/frameworks/keras_pipe_mode_horovod/generate_cifar10_tfrecords.py
diff --git a/aws_sagemaker_studio/frameworks/keras_pipe_mode_horovod/generate_tensorboard_command.py b/aws_sagemaker_studio/frameworks/keras_pipe_mode_horovod/generate_tensorboard_command.py