Skip to content

Commit 661890c

Browse files
committed
lora: Fix training start from step 2
1 parent 082e7a0 commit 661890c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/llama-context.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2290,7 +2290,7 @@ void llama_context::opt_epoch(
22902290
std::vector<llama_token> labels_sparse(n_ctx);
22912291
std::vector<int32_t> masks_sparse(n_ctx);
22922292

2293-
int64_t idata = (resume_from_batch >= 0) ? resume_from_batch + 1 : 0;
2293+
int64_t idata = (resume_from_batch > 0) ? resume_from_batch + 1 : 0;
22942294

22952295
int64_t t_loop_start = ggml_time_us();
22962296
int64_t ndata_in_loop = idata_split*ubatch_per_ctx;

0 commit comments

Comments
 (0)