Update train_a_generative_llm.py

Print shapes for diagnostics.

Author: david-thrower

train_a_generative_llm.py

@@ -254,8 +254,10 @@
 print(f"Shape of y_test: {len(y_test)}")
 
 x_train_tf = tf.constant(X_train, tf.int32)
+print(x_train_tf)
 y_train_tf = tf.constant(y_train, tf.float32)
-        
+print(y_train_tf)
+
 x_train_packaged = [x_train_tf]
 y_train_packaged = [y_train_tf]