Skip to content

README.md

Latest commit

 

History

History
14 lines (11 loc) · 1020 Bytes

README.md

File metadata and controls

14 lines (11 loc) · 1020 Bytes

ETL data on CPU and simultaneously Training model on GPU using TFDS

We are going to use CPU for Extract , Transform and Load data from Tensorflow Dataset, and parallelly GPU for training model parallelly.

The code is available for the same in notebook file present in this notebook.

  • This code will not work on Google Colab as you cannot use both cpu and gpu in colab.
  • Make sure you are running this code on Machine containing both CPU and GPU and tensorflow with GPU support. You can find tensorflow gpu installation documentation here.
  • If you are getting
ValueError: No data provided for "input_1". Need data for each key in: ['input_1']

Then you are either running code on google colab or on a machine which has either cpu or gpu only.

You can also refer my blog from here.