Update train.py

llmshearing/train.py

@@ -32,6 +32,7 @@
 from llmshearing.datasets.load_text_dataloader import build_text_dataloader
 from llmshearing.models.model_registry import COMPOSER_MODEL_REGISTRY
 
+streaming.base.util.clean_stale_shared_memory()
 
 def is_one_hour(run_name: str):
     """ Check if the run name is for one hour training. """