Executing Eager mode - the machine is not responding after execution

[OrielResearchCure]

Hi all,

I am running the following code:

!pip install tensorflow --upgrade
import tensorflow as tf
tf.enable_eager_execution()

CSV_PATH_TRAIN = '/path/to/csv/file.csv'
dataset = tf.data.experimental.make_csv_dataset(file_pattern=CSV_PATH_TRAIN,batch_size=1,header=True,label_name = 'label').map(load_image)

for batch in tfe.Iterator(dataset):
     print(batch)

The notebook is running the code and hanging. The only way to be able to keep in working is killing the notebook

Please let me know what is the right way to run eager.

Thanks, eilalan