We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3bdf63c commit 446ad8eCopy full SHA for 446ad8e
QEfficient/finetune/utils/train_utils.py
@@ -178,6 +178,7 @@ def train(
178
# adjust atol & rtol this as required
179
atol=1e-1,
180
use_ref_output_on_mismatch=True,
181
+ filter_config=qaic_debug.DispatchFilterConfig.default(device),
182
dump_root_dir=train_config.dump_root_dir + str(step),
183
) as verifier:
184
loss = model(**batch).loss # Forward call
0 commit comments