From 11aff5dde388f67c348b41ad1de3222da8e1e09b Mon Sep 17 00:00:00 2001 From: Juhan Bae Date: Wed, 20 Mar 2024 03:46:06 -0400 Subject: [PATCH] Lower SVD precision --- examples/imagenet/analyze.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/imagenet/analyze.py b/examples/imagenet/analyze.py index 677f49d..418be1c 100644 --- a/examples/imagenet/analyze.py +++ b/examples/imagenet/analyze.py @@ -136,7 +136,7 @@ def main(): train_dataset=train_dataset, per_device_query_batch_size=args.query_batch_size, per_device_train_batch_size=args.train_batch_size, - overwrite_output_dir=False, + overwrite_output_dir=True, )