You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I am trying to train a TensorFlow custom classification model using a custom data generator on SageMaker Studio. However, the notebook did not utilize the GPU to train the model.
My instance type:
I have stored the images in a zip file on an S3 bucket and unzipped them during training, using the temporary path in the custom generator for training.
I want to add data augmentation while training the model.
The example below loads the images in numpy format and passes them directly to the training model. I want to load the images from the S3 bucket in zip format and train with a custom generator and augmentation.
The text was updated successfully, but these errors were encountered:
dsnsabari
changed the title
How to use the custom generator for training the tensorflow model on sagemkaer
How do you use the custom generator to train the TensorFlow model on PageMaker?
Feb 27, 2024
I am trying to train a TensorFlow custom classification model using a custom data generator on SageMaker Studio. However, the notebook did not utilize the GPU to train the model.
My instance type:
I have stored the images in a zip file on an S3 bucket and unzipped them during training, using the temporary path in the custom generator for training.
I want to add data augmentation while training the model.
The example below loads the images in numpy format and passes them directly to the training model. I want to load the images from the S3 bucket in zip format and train with a custom generator and augmentation.
Link to the example
The text was updated successfully, but these errors were encountered: