zetavg commited on
Commit
2adc4d2
2 Parent(s): c9cf2f1 dfc944d

Merge branch 'main' into hf-ui-demo

Browse files
Files changed (1) hide show
  1. llama_lora/ui/finetune/training.py +4 -2
llama_lora/ui/finetune/training.py CHANGED
@@ -216,6 +216,8 @@ def do_train(
216
 
217
  total_epochs = epochs
218
  total_steps = len(train_data) * epochs
 
 
219
  log_history = []
220
  initial_loss = 2
221
  loss_decay_rate = 0.8
@@ -242,11 +244,11 @@ def do_train(
242
  current_epoch=current_epoch,
243
  log_history=log_history
244
  )
245
- time.sleep(0.1)
246
 
247
  result_message = set_train_output(message)
248
  print(result_message)
249
- time.sleep(1)
250
  Global.is_training = False
251
  return
252
 
 
216
 
217
  total_epochs = epochs
218
  total_steps = len(train_data) * epochs
219
+ if total_steps < 1500:
220
+ total_steps = 1500
221
  log_history = []
222
  initial_loss = 2
223
  loss_decay_rate = 0.8
 
244
  current_epoch=current_epoch,
245
  log_history=log_history
246
  )
247
+ time.sleep(0.01)
248
 
249
  result_message = set_train_output(message)
250
  print(result_message)
251
+ time.sleep(3)
252
  Global.is_training = False
253
  return
254