Skip to content

Commit

Permalink
Fix a seed for a unit test to dig into whether it is something else.
Browse files Browse the repository at this point in the history
  • Loading branch information
liuliu committed Jun 10, 2024
1 parent 6b53fc6 commit 7f464ef
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions test/unit/nnc/cnnp.core.tests.c
Original file line number Diff line number Diff line change
Expand Up @@ -2072,6 +2072,7 @@ TEST_CASE("LoRA fine-tuning GEMM set is_trainable to false and with gradient che

TEST_CASE("LoRA fine-tuning MLP with GELU, set is_trainable to false and with gradient checkpointing")
{
ccv_nnc_stream_context_set_seed(0, 47);
const ccv_cnnp_model_io_t input = ccv_cnnp_input();
ccv_cnnp_model_t* const fc1 = ccv_cnnp_dense(10, 1, -1, "fc1");
ccv_cnnp_model_t* const fc2 = ccv_cnnp_dense(10, 1, -1, "fc2");
Expand Down

0 comments on commit 7f464ef

Please sign in to comment.