Skip to content

Commit 8cff2fe

Browse files
authored
update Batch_Size to 128
1 parent 7a62220 commit 8cff2fe

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

config.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ def get_arguments():
2121
parser.add_argument('--finetuning_ascent_model', type=str, default=True, help='whether finetuning model')
2222
parser.add_argument('--finetuning_epochs', type=int, default=60, help='number of finetuning epochs to run')
2323
parser.add_argument('--unlearning_epochs', type=int, default=5, help='number of unlearning epochs to run')
24-
parser.add_argument('--batch_size', type=int, default=64, help='The size of batch')
24+
parser.add_argument('--batch_size', type=int, default=128, help='The size of batch')
2525
parser.add_argument('--lr', type=float, default=0.1, help='initial learning rate')
2626
parser.add_argument('--lr_finetuning_init', type=float, default=0.1, help='initial finetuning learning rate')
2727
parser.add_argument('--lr_unlearning_init', type=float, default=5e-4, help='initial unlearning learning rate')

0 commit comments

Comments
 (0)