modified an argument type

Signed-off-by: ftgreat <[email protected]>
SysuJayce · Jul 3, 2023 · 184c2d3 · 184c2d3
1 parent 95158b9
commit 184c2d3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flagai/env_args.py b/flagai/env_args.py
@@ -165,7 +165,7 @@ def __init__(self,
         self.parser.add_argument('--bmt_loss_scale', default=bmt_loss_scale, type=float, help='loss scale in bmtrain')
         self.parser.add_argument('--bmt_loss_scale_steps', default=bmt_loss_scale_steps, type=int, help='loss scale steps in bmtrain')
 
-        self.parser.add_argument('--lora', default=lora, type=bool, help='Use lora')
+        self.parser.add_argument('--lora', default=lora, type=str2bool, help='Use lora')
         self.parser.add_argument('--lora_r', default=lora_r, typp=int, help='lora r value')
         self.parser.add_argument('--lora_alpha', default=lora_alpha, type=float, help='lora alpha value')
         self.parser.add_argument('--lora_dropout', default=lora_dropout, type=float, help='lora dropout value')