Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
zetavg committed Apr 27, 2023
1 parent fcc807e commit dfc944d
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions llama_lora/ui/finetune/training.py
Original file line number Diff line number Diff line change
Expand Up @@ -216,6 +216,8 @@ def training():

total_epochs = epochs
total_steps = len(train_data) * epochs
if total_steps < 1500:
total_steps = 1500
log_history = []
initial_loss = 2
loss_decay_rate = 0.8
Expand All @@ -242,11 +244,11 @@ def training():
current_epoch=current_epoch,
log_history=log_history
)
time.sleep(0.1)
time.sleep(0.01)

result_message = set_train_output(message)
print(result_message)
time.sleep(1)
time.sleep(3)
Global.is_training = False
return

Expand Down

0 comments on commit dfc944d

Please sign in to comment.