Update_Readme

OliverRensu · Dec 5, 2023 · 5ca2a95 · 5ca2a95
1 parent 26bf29f
commit 5ca2a95
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/DiGPT_torch/README.md b/DiGPT_torch/README.md
@@ -36,7 +36,7 @@ python -m torch.distributed.launch --nproc_per_node=8 main_finetune.py \
     --finetune /path/to/checkpoint-299.pth \
     --epochs 100 \
     --output_dir ./out_finetune/ \
-    --blr 1e-4 --layer_decay 0.6 \
+    --blr 1e-4 --layer_decay 0.65 \
     --weight_decay 0.05 --drop_path 0.1 --reprob 0.25 --mixup 0.8 --cutmix 1.0 \
     --dist_eval --data_path /path/to/ImageNet/
 ```