Update examples/language_modeling.py

Lightning-Universe · May 31, 2022 · 7a0e982 · 7a0e982
1 parent 521e6db
commit 7a0e982
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/language_modeling.py b/examples/language_modeling.py
@@ -14,6 +14,6 @@
         cfg=LanguageModelingDataConfig(batch_size=1, dataset_name="wikitext", dataset_config_name="wikitext-2-raw-v1"),
         tokenizer=tokenizer,
     )
-    trainer = pl.Trainer(accelerator="auto", devices="auto")
+    trainer = pl.Trainer(accelerator="auto", devices="auto", max_epochs=1)
 
     trainer.fit(model, dm)