Skip to content

Commit 8e21fe9

Browse files
committed
Need to bring back filter_config due to change in arguments
Signed-off-by: Meet Patel <[email protected]>
1 parent 998f7bb commit 8e21fe9

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

QEfficient/finetune/utils/train_utils.py

+1
Original file line numberDiff line numberDiff line change
@@ -177,6 +177,7 @@ def train(
177177
# adjust atol & rtol this as required
178178
atol=1e-1,
179179
use_ref_output_on_mismatch=True,
180+
filter_config=qaic_debug.DispatchFilterConfig.default(device),
180181
dump_root_dir=train_config.dump_root_dir + str(step),
181182
) as verifier:
182183
loss = model(**batch).loss # Forward call

0 commit comments

Comments
 (0)