Skip to content

Commit 446ad8e

Browse files
committed
Need to bring back filter_config due to change in arguments
Signed-off-by: Meet Patel <[email protected]>
1 parent 3bdf63c commit 446ad8e

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

QEfficient/finetune/utils/train_utils.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -178,6 +178,7 @@ def train(
178178
# adjust atol & rtol this as required
179179
atol=1e-1,
180180
use_ref_output_on_mismatch=True,
181+
filter_config=qaic_debug.DispatchFilterConfig.default(device),
181182
dump_root_dir=train_config.dump_root_dir + str(step),
182183
) as verifier:
183184
loss = model(**batch).loss # Forward call

0 commit comments

Comments
 (0)