diff --git a/finetune/lora/train.py b/finetune/lora/train.py index a03322f..78f2646 100644 --- a/finetune/lora/train.py +++ b/finetune/lora/train.py @@ -264,7 +264,7 @@ if __name__ == "__main__": # # Data = {args.data_file} ({args.data_type}), ProjDir = {args.proj_dir} # -# Epoch = {args.epoch_begin} to {args.epoch_begin + args.epoch_count - 1} (will continue afterwards), save every {args.epoch_save} epoch +# Epoch = {args.epoch_begin} to {args.epoch_begin + args.epoch_count - 1}, save every {args.epoch_save} epoch # # Each "epoch" = {args.epoch_steps} steps, {samples_per_epoch} samples, {tokens_per_epoch} tokens # diff --git a/frontend/src/stores/commonStore.ts b/frontend/src/stores/commonStore.ts index d720807..08bf0c1 100644 --- a/frontend/src/stores/commonStore.ts +++ b/frontend/src/stores/commonStore.ts @@ -102,7 +102,7 @@ class CommonStore { epochSteps: 200, epochCount: 20, epochBegin: 0, - epochSave: 2, + epochSave: 1, microBsz: 1, accumGradBatches: 8, preFfn: false,