new file: Figure_1.png

deleted:    best_model.pth
	new file:   checkpoints/forward/best_tcn_model.pt
	new file:   checkpoints/forward/training_history.csv
	new file:   evaluation_outputs/forward/evaluation_forward_test_b0_s0.png
	new file:   sanity_check_alignment_forward.png
	modified:   src/__pycache__/config.cpython-310.pyc
	modified:   src/__pycache__/config.cpython-314.pyc
	modified:   src/__pycache__/dataset.cpython-310.pyc
	modified:   src/__pycache__/dataset.cpython-314.pyc
	modified:   src/__pycache__/model.cpython-310.pyc
	modified:   src/__pycache__/model.cpython-314.pyc
	modified:   src/config.py
	modified:   src/dataset.py
	modified:   src/evaluate.py
	modified:   src/model.py
	new file:   src/sanity_check.py
	modified:   src/train.py
	renamed:    src/__init__.py -> src_old/__init__.py
	new file:   src_old/__pycache__/config.cpython-310.pyc
	new file:   src_old/__pycache__/config.cpython-314.pyc
	new file:   src_old/__pycache__/dataset.cpython-310.pyc
	new file:   src_old/__pycache__/dataset.cpython-314.pyc
	new file:   src_old/__pycache__/evaluate.cpython-314.pyc
	new file:   src_old/__pycache__/model.cpython-310.pyc
	new file:   src_old/__pycache__/model.cpython-314.pyc
	new file:   src_old/__pycache__/train.cpython-310.pyc
	new file:   src_old/__pycache__/train.cpython-314.pyc
	new file:   src_old/config.py
	new file:   src_old/dataset.py
	new file:   src_old/evaluate.py
	new file:   src_old/model.py
	new file:   src_old/train.py
	deleted:    test_results.png
This commit is contained in:
2026-05-04 18:18:44 +08:00
parent c65e47e6b0
commit dcc023cc04
34 changed files with 2177 additions and 436 deletions

View File

@@ -0,0 +1,14 @@
epoch,lr,train_loss,train_time_loss,train_weighted_time_loss,train_fft_loss,train_rms_loss,train_scale_loss,train_underestimate_loss,train_envelope_loss,train_rms_error,train_dominant_freq_error,val_loss,val_time_loss,val_weighted_time_loss,val_fft_loss,val_rms_loss,val_scale_loss,val_underestimate_loss,val_envelope_loss,val_rms_error,val_dominant_freq_error
1,0.001,32.23490004269582,0.9063257234838774,0.09063257336757093,2462.537229142099,0.5202015545570625,0.5072746667659508,0.23183272828189833,0.9878060196368199,0.5202015545570625,0.6667674588373308,16.848691019518622,0.5030307245665583,0.05030307315033058,712.1304384428879,0.21638367042459292,0.3054182426682834,0.16712227438030572,0.7986705179872184,0.21638367042459292,0.9571996228951049
2,0.001,21.43670712776904,0.9317702313639084,0.09317702508338217,1694.0047123747052,0.35049294160222105,0.3616391454102858,0.1121402242625097,0.672775794312639,0.35049294160222105,0.6002240721274537,18.15417513354071,0.5811605972462687,0.058116060392609956,948.0657793242356,0.23726162314414978,0.3922087324076685,0.15394838554142365,0.7320991713425209,0.23726162314414978,0.8361816405868994
3,0.001,18.081004331696708,0.949055330933265,0.09490553458344261,1372.9708672289578,0.28145609332143134,0.3131425976753235,0.09374795892750318,0.6080609613432074,0.28145609332143134,0.616063318727439,16.73266759412042,0.5420598767954727,0.05420598895128431,673.0913622625943,0.238979727286717,0.33492567107595245,0.1804515516449665,0.7293170896069757,0.238979727286717,0.7597824622933206
4,0.001,15.158426779621053,0.9718681652590914,0.09718681784030402,1005.0683164776497,0.2189214069325969,0.2836238658934269,0.08367646379695046,0.5845890998278024,0.2189214069325969,0.6179393338571753,17.278400026518725,0.5685602380283947,0.056856024560743366,892.716255977236,0.22618299260221678,0.36168746264844104,0.1538948353444194,0.6990852653980255,0.22618299260221678,0.7597824622933206
5,0.001,13.61913053044733,0.9698394334541177,0.09698394513776842,859.7200766509434,0.19106561557020782,0.2557676170232161,0.07612784807833861,0.5538543634257227,0.19106561557020782,0.6155450957215706,17.31632186626566,0.5785790409507423,0.057857905097048856,895.00206572434,0.21634554451909557,0.35666010605877846,0.1514351657302729,0.7220557270378902,0.21634554451909557,0.7402091190436203
6,0.001,12.678725962368947,0.9694307513956754,0.0969430767351164,788.4346546676924,0.1713324544845887,0.23463592369039105,0.07080332469195127,0.5323761788741598,0.1713324544845887,0.6164817462027068,18.423015594482422,0.5721222347226637,0.057212224808232535,888.629751271215,0.25913861959144985,0.4007708651238474,0.16395779438959113,0.7525067822686557,0.25913861959144985,0.7218985721265917
7,0.001,12.268796533908484,0.967802310889622,0.09678023287429,744.56112426182,0.1675024907684551,0.2299804154713199,0.06777841198029665,0.5231116253812358,0.1675024907684551,0.5961900787542253,17.48041952067408,0.5510032737049563,0.055100328578003524,704.0817605380354,0.2379121222886546,0.37086750232967836,0.19612120528673305,0.7370918125941835,0.2379121222886546,0.7088496766840734
8,0.001,11.889431161700555,0.967772865070487,0.09677728824317455,712.115890215028,0.1614244244289848,0.2226250537161557,0.06655820054968573,0.5118434342010966,0.1614244244289848,0.6022199021486582,18.410391971982758,0.5461424337378864,0.05461424415738418,738.1983037488214,0.2533088120920905,0.3795644687167529,0.21159803308546543,0.7770289305982918,0.2533088120920905,0.707797346401937
9,0.001,11.234980547203207,0.9654024828155086,0.09654024967326308,670.3140682004532,0.14827435323089924,0.20859187953876998,0.062428659900038874,0.4916352005499714,0.14827435323089924,0.5623616312479263,16.51186601046858,0.5469307437025267,0.05469307598882708,658.5074381335028,0.23887184570575581,0.3477302913008065,0.18191729235494958,0.6951427665250055,0.23887184570575581,0.7065345500594454
10,0.001,10.992490152143082,0.9626774315564137,0.09626774419591112,649.039663350807,0.14625250653557056,0.20260991113928128,0.061842711396374796,0.48383018318212256,0.14625250653557056,0.568747647623218,17.501161114922887,0.5506949491541961,0.05506949608439002,671.2075416301859,0.2541997463538729,0.373840541675173,0.19259302314884705,0.7480303932880533,0.2541997463538729,0.708428744563363
11,0.0005,10.806822551871246,0.9680146614335617,0.09680146787245318,637.6420590382702,0.14451587228280194,0.2035975716305229,0.0572100906404403,0.4776927213061531,0.14451587228280194,0.5574953460125556,17.13407661174906,0.5288547395632185,0.05288547495829648,618.1973874322299,0.2417878899081,0.34800343832065317,0.21005815672206468,0.7388678706925491,0.2417878899081,0.7122171335564683
12,0.0005,10.255595112746617,0.9616962041494981,0.09616962194723903,599.1373590433373,0.12593858316540718,0.18720266560338578,0.05598305191247249,0.4703599948365733,0.12593858316540718,0.5670544286673502,16.41022307297279,0.5580467443014013,0.055804675894564594,691.4893851444639,0.23402822017669678,0.3565609496215294,0.17255781141334567,0.6764166149599798,0.23402822017669678,0.6966426454535165
13,0.0005,9.979247138185322,0.9624017550135558,0.09624017732885648,580.7163422782467,0.12185739905063836,0.18553815514973873,0.05258095577657926,0.4598706980358879,0.12185739905063836,0.5524671217591512,16.273903156148975,0.5295915886245924,0.05295915985158805,541.1639246447332,0.23793373128463483,0.3361044392503541,0.20454522344315873,0.7062619534032099,0.23793373128463483,0.7056926858376642
1 epoch lr train_loss train_time_loss train_weighted_time_loss train_fft_loss train_rms_loss train_scale_loss train_underestimate_loss train_envelope_loss train_rms_error train_dominant_freq_error val_loss val_time_loss val_weighted_time_loss val_fft_loss val_rms_loss val_scale_loss val_underestimate_loss val_envelope_loss val_rms_error val_dominant_freq_error
2 1 0.001 32.23490004269582 0.9063257234838774 0.09063257336757093 2462.537229142099 0.5202015545570625 0.5072746667659508 0.23183272828189833 0.9878060196368199 0.5202015545570625 0.6667674588373308 16.848691019518622 0.5030307245665583 0.05030307315033058 712.1304384428879 0.21638367042459292 0.3054182426682834 0.16712227438030572 0.7986705179872184 0.21638367042459292 0.9571996228951049
3 2 0.001 21.43670712776904 0.9317702313639084 0.09317702508338217 1694.0047123747052 0.35049294160222105 0.3616391454102858 0.1121402242625097 0.672775794312639 0.35049294160222105 0.6002240721274537 18.15417513354071 0.5811605972462687 0.058116060392609956 948.0657793242356 0.23726162314414978 0.3922087324076685 0.15394838554142365 0.7320991713425209 0.23726162314414978 0.8361816405868994
4 3 0.001 18.081004331696708 0.949055330933265 0.09490553458344261 1372.9708672289578 0.28145609332143134 0.3131425976753235 0.09374795892750318 0.6080609613432074 0.28145609332143134 0.616063318727439 16.73266759412042 0.5420598767954727 0.05420598895128431 673.0913622625943 0.238979727286717 0.33492567107595245 0.1804515516449665 0.7293170896069757 0.238979727286717 0.7597824622933206
5 4 0.001 15.158426779621053 0.9718681652590914 0.09718681784030402 1005.0683164776497 0.2189214069325969 0.2836238658934269 0.08367646379695046 0.5845890998278024 0.2189214069325969 0.6179393338571753 17.278400026518725 0.5685602380283947 0.056856024560743366 892.716255977236 0.22618299260221678 0.36168746264844104 0.1538948353444194 0.6990852653980255 0.22618299260221678 0.7597824622933206
6 5 0.001 13.61913053044733 0.9698394334541177 0.09698394513776842 859.7200766509434 0.19106561557020782 0.2557676170232161 0.07612784807833861 0.5538543634257227 0.19106561557020782 0.6155450957215706 17.31632186626566 0.5785790409507423 0.057857905097048856 895.00206572434 0.21634554451909557 0.35666010605877846 0.1514351657302729 0.7220557270378902 0.21634554451909557 0.7402091190436203
7 6 0.001 12.678725962368947 0.9694307513956754 0.0969430767351164 788.4346546676924 0.1713324544845887 0.23463592369039105 0.07080332469195127 0.5323761788741598 0.1713324544845887 0.6164817462027068 18.423015594482422 0.5721222347226637 0.057212224808232535 888.629751271215 0.25913861959144985 0.4007708651238474 0.16395779438959113 0.7525067822686557 0.25913861959144985 0.7218985721265917
8 7 0.001 12.268796533908484 0.967802310889622 0.09678023287429 744.56112426182 0.1675024907684551 0.2299804154713199 0.06777841198029665 0.5231116253812358 0.1675024907684551 0.5961900787542253 17.48041952067408 0.5510032737049563 0.055100328578003524 704.0817605380354 0.2379121222886546 0.37086750232967836 0.19612120528673305 0.7370918125941835 0.2379121222886546 0.7088496766840734
9 8 0.001 11.889431161700555 0.967772865070487 0.09677728824317455 712.115890215028 0.1614244244289848 0.2226250537161557 0.06655820054968573 0.5118434342010966 0.1614244244289848 0.6022199021486582 18.410391971982758 0.5461424337378864 0.05461424415738418 738.1983037488214 0.2533088120920905 0.3795644687167529 0.21159803308546543 0.7770289305982918 0.2533088120920905 0.707797346401937
10 9 0.001 11.234980547203207 0.9654024828155086 0.09654024967326308 670.3140682004532 0.14827435323089924 0.20859187953876998 0.062428659900038874 0.4916352005499714 0.14827435323089924 0.5623616312479263 16.51186601046858 0.5469307437025267 0.05469307598882708 658.5074381335028 0.23887184570575581 0.3477302913008065 0.18191729235494958 0.6951427665250055 0.23887184570575581 0.7065345500594454
11 10 0.001 10.992490152143082 0.9626774315564137 0.09626774419591112 649.039663350807 0.14625250653557056 0.20260991113928128 0.061842711396374796 0.48383018318212256 0.14625250653557056 0.568747647623218 17.501161114922887 0.5506949491541961 0.05506949608439002 671.2075416301859 0.2541997463538729 0.373840541675173 0.19259302314884705 0.7480303932880533 0.2541997463538729 0.708428744563363
12 11 0.0005 10.806822551871246 0.9680146614335617 0.09680146787245318 637.6420590382702 0.14451587228280194 0.2035975716305229 0.0572100906404403 0.4776927213061531 0.14451587228280194 0.5574953460125556 17.13407661174906 0.5288547395632185 0.05288547495829648 618.1973874322299 0.2417878899081 0.34800343832065317 0.21005815672206468 0.7388678706925491 0.2417878899081 0.7122171335564683
13 12 0.0005 10.255595112746617 0.9616962041494981 0.09616962194723903 599.1373590433373 0.12593858316540718 0.18720266560338578 0.05598305191247249 0.4703599948365733 0.12593858316540718 0.5670544286673502 16.41022307297279 0.5580467443014013 0.055804675894564594 691.4893851444639 0.23402822017669678 0.3565609496215294 0.17255781141334567 0.6764166149599798 0.23402822017669678 0.6966426454535165
14 13 0.0005 9.979247138185322 0.9624017550135558 0.09624017732885648 580.7163422782467 0.12185739905063836 0.18553815514973873 0.05258095577657926 0.4598706980358879 0.12185739905063836 0.5524671217591512 16.273903156148975 0.5295915886245924 0.05295915985158805 541.1639246447332 0.23793373128463483 0.3361044392503541 0.20454522344315873 0.7062619534032099 0.23793373128463483 0.7056926858376642