From 295af9a5174944d4747b57f175573606c746b8f9 Mon Sep 17 00:00:00 2001 From: BlinkDL Date: Wed, 18 Jan 2023 16:23:54 +0000 Subject: [PATCH] info --- RWKV-v4neo/train.py | 1 + 1 file changed, 1 insertion(+) diff --git a/RWKV-v4neo/train.py b/RWKV-v4neo/train.py index 2b2d3e8..fd90b10 100644 --- a/RWKV-v4neo/train.py +++ b/RWKV-v4neo/train.py @@ -100,6 +100,7 @@ if __name__ == "__main__": parser.add_argument("--my_pos_emb", default=0, type=int) parser.add_argument("--load_partial", default=0, type=int) parser.add_argument("--magic_prime", default=0, type=int) + parser.add_argument("--my_testing", default=0, type=int) parser = Trainer.add_argparse_args(parser) args = parser.parse_args()