Michael Diskin 4 lat temu
rodzic
commit
0fa9c46f68
1 zmienionych plików z 2 dodań i 2 usunięć
  1. 2 2
      examples/albert/arguments.py

+ 2 - 2
examples/albert/arguments.py

@@ -86,7 +86,7 @@ class CollaborativeOptimizerArguments:
         metadata={"help": "Available network bandwidth, in mbps (used for load balancing in all-reduce)"}
     )
     compression: str = field(
-        default="FLOAT16",
+        default="NONE",
         metadata={"help": "Use this compression when averaging parameters/gradients"}
     )
 
@@ -139,7 +139,7 @@ class AlbertTrainingArguments(TrainingArguments):
     max_grad_norm: float = 1.0
     clamp_value: float = 10000.0
 
-    fp16: bool = True
+    fp16: bool = False
     fp16_opt_level: str = 'O2'
     do_train: bool = True