diff --git a/model/model-0.91502.jld2 b/model/model-0.91502.jld2 deleted file mode 100644 index 90edf20..0000000 Binary files a/model/model-0.91502.jld2 and /dev/null differ diff --git a/model/model-0.92091.jld2 b/model/model-0.92091.jld2 new file mode 100644 index 0000000..2758b89 Binary files /dev/null and b/model/model-0.92091.jld2 differ diff --git a/src/data_processing.jl b/src/data_processing.jl index dfe9f01..1ec9cfd 100644 --- a/src/data_processing.jl +++ b/src/data_processing.jl @@ -98,7 +98,7 @@ Predict left hand or right hand by output. predict_y(y) = y > 0.5f0 ? 1 : 0 function generate_midi(input_file::String; output_file::String="", - weight_file=pkgdir(PianoHands,"model","model-0.91502.jld2"),HIDDEN_SIZE=14) + weight_file=pkgdir(PianoHands,"model","model-0.92091.jld2"),HIDDEN_SIZE=14) midi_file = load(input_file) hand_classify = inferance_midi(midi_file,weight_file,HIDDEN_SIZE) diff --git a/src/training.jl b/src/training.jl index 4066b9d..5ac4593 100644 --- a/src/training.jl +++ b/src/training.jl @@ -8,7 +8,7 @@ end function train_piano(DATASET_PATH, TESTSET_PATH; BATCH_SIZE = 12, - SEQ_LENGTH = 75, + SEQ_LENGTH = 65, HIDDEN_SIZE = 14, LEARNING_RATE = 0.0005f0, MAX_EPOCH = 200, diff --git a/test/runtests.jl b/test/runtests.jl index cb680ac..83c94b3 100644 --- a/test/runtests.jl +++ b/test/runtests.jl @@ -3,12 +3,12 @@ using Test,PianoHands,MIDI,Lux,Random,Printf,LuxCUDA,Optimisers,Zygote,JLD2 @testset "pig to feature" begin # train_piano("../PianoFingeringDataset/dataset/", # "../PianoFingeringDataset/testset/"; - # SEQ_LENGTH=75, + # SEQ_LENGTH=65, # BATCH_SIZE=12, # LEARNING_RATE = 0.0005f0, # HIDDEN_SIZE = 14, # EVALUATE_PER_N_TRAIN = 50 # ) - generate_midi("./ymsn_full.mid";weight_file="../model/model-0.91502.jld2") + generate_midi("./ymsn_full.mid";weight_file="../model/model-0.92091.jld2") end \ No newline at end of file