[HPC_AI][AI for Science Climate] Improvements

[mozhgan-kch]

Suggestions related to the cell block in Resnets notebook and CNN:

# Reshape input data from (28, 28) to (28, 28, 1)
# w, h = 28, 28
#train_images = train_images.reshape(train_images.shape[0], w, h)
#test_images = test_images.reshape(test_images.shape[0], w, h)

Deletion of entire lines in the "Making predictions" block. It overwrites train_image on test_image variable names and prevents the previous cells to be re-run again. And we are not using "train_images" in the following cells.
This back-conversion is necessary?