We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 998f7bb commit 8e21fe9Copy full SHA for 8e21fe9
QEfficient/finetune/utils/train_utils.py
@@ -177,6 +177,7 @@ def train(
177
# adjust atol & rtol this as required
178
atol=1e-1,
179
use_ref_output_on_mismatch=True,
180
+ filter_config=qaic_debug.DispatchFilterConfig.default(device),
181
dump_root_dir=train_config.dump_root_dir + str(step),
182
) as verifier:
183
loss = model(**batch).loss # Forward call
0 commit comments