Tutorial 3 Colab Link : https://colab.research.google.com/drive/1XfT75glapWdCp0Zj_4mkiMrjT6odD-Gc
Tutorial 5 Colab Link : https://colab.research.google.com/drive/13lBsht3Wa4GjKKkA47JCrd54XikhNX2E

I am facing Out Of Memory(OOM) while allocating tensor during training of classifier in Tutorial 5. I've tried 12 times and all of the time, it's happening after Epoch 4 out of 100.
Further, I couldn't get the packages to install. The error I am getting is 'toolchain is invalid'
It all started happening after S4TF update to v0.4

Following are the screenshots of the issue regarding OOM


Screenshot regarding Package Install

The cats vs dogs dataset in this repo, should be removed. Instead provide a link to download that dataset

Tracker so that we don't forget the task.

With Swift for TensorFlow, and the Swift data science ecosystem in general, still being fairly early-stage, many elements of Python-based TensorFlow don't have equivalents on the Swift side. This looks like it might be an increasing issue with converting tutorials going forward, so it might be helpful to first list off libraries or capabilities that tutorials depend on but that don't exist yet.

I figured having an open issue like this might be a good way to track and discuss these, although we might be able to work this into a GitHub Project as well.

Here's the list that we've discussed so far, and feel free to add others:

• tensorflow_datasets -> would be great to leverage the existing work done here to make it easy to download / set up datasets
• image loading (tf.data, PIL / Pillow)
• simple image manipulation for augmentation (tensorflow.keras.preprocessing.image.ImageDataGenerator): flipping, shear, translation, cropping
• loading pretrained weights (TensorFlow Hub): Needs SavedModel loading, can currently pull from checkpoints via Raw TensorFlow operators