Skip to content

Commit 41344eb

Browse files
committed
disable 16mixed precision bits, it causes issues with na loss values
1 parent f501af5 commit 41344eb

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

flexynesis/main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -154,7 +154,7 @@ def setup_trainer(self, params, current_step, total_steps, full_train = False):
154154

155155
trainer = pl.Trainer(
156156
#deterministic = True,
157-
precision = '16-mixed', # mixed precision training
157+
#precision = '16-mixed', # mixed precision training
158158
max_epochs=int(params['epochs']),
159159
gradient_clip_val=1.0,
160160
gradient_clip_algorithm='norm',

0 commit comments

Comments
 (0)