added todo

train.py

@@ -12,6 +12,8 @@ from pytorch_lightning import Trainer
 import torch
 
 
+# TODO: make label smoothing scale with random magnitude, generate photorealistic images with notebook background
+
 def check_setup():
     # Disabling tokenizers parallelism because it can't be used before forking and I didn't bother to figure it out
     os.environ["TOKENIZERS_PARALLELISM"] = "false"