diff --git a/ray_gpt.py b/ray_gpt.py index 70c2a5b64fc32608c832e1db208d2c1d9bcbd199..cc01ce36b17d5ce86fb5ec1b294f747dad6ee142 100644 --- a/ray_gpt.py +++ b/ray_gpt.py @@ -36,7 +36,7 @@ def main(config): def main_task(config): trainer = get_trainer(config.training.stage)(config) trainer.train() - +# test if __name__ == '__main__': main() \ No newline at end of file