Add files using upload-large-folder tool
Browse files- README.md +13 -0
- best_model.pth +3 -0
- config.json +12 -0
- final_model.pth +3 -0
- training_history.json +314 -0
README.md
ADDED
|
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: mit
|
| 3 |
+
tags:
|
| 4 |
+
- aquatrace
|
| 5 |
+
- plastic-detection
|
| 6 |
+
- ocean
|
| 7 |
+
---
|
| 8 |
+
|
| 9 |
+
# aquatrace-drift-simulation
|
| 10 |
+
|
| 11 |
+
AquaTrace – Module 4: Ocean Plastic Drift Simulation
|
| 12 |
+
|
| 13 |
+
Part of the **AquaTrace** AI pipeline for marine plastic monitoring.
|
best_model.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:31a9bf4f291986232f4490670e0e7c20089eee9b336084528551b1f9b1af616e
|
| 3 |
+
size 4835182
|
config.json
ADDED
|
@@ -0,0 +1,12 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"nc_file": "cmems_mod_glo_phy_anfc_0.083deg_P1D-m_1772194204759.nc",
|
| 3 |
+
"num_particles": 5000,
|
| 4 |
+
"trajectory_length": 24,
|
| 5 |
+
"batch_size": 32,
|
| 6 |
+
"num_epochs": 50,
|
| 7 |
+
"lr": 0.0001,
|
| 8 |
+
"num_workers": 4,
|
| 9 |
+
"hidden_size": 256,
|
| 10 |
+
"num_layers": 6,
|
| 11 |
+
"device": "cuda"
|
| 12 |
+
}
|
final_model.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4a6b4fd988f6b0d757952bd0aad3454d0e213eb2b489b69291500958e3c34c96
|
| 3 |
+
size 4835426
|
training_history.json
ADDED
|
@@ -0,0 +1,314 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"train_loss": [
|
| 3 |
+
984.9998956853693,
|
| 4 |
+
56.99208969636397,
|
| 5 |
+
15.34129819869995,
|
| 6 |
+
8.034396895495329,
|
| 7 |
+
6.142068715529009,
|
| 8 |
+
4.768390057303689,
|
| 9 |
+
5.1034928755326705,
|
| 10 |
+
3.4052254059097984,
|
| 11 |
+
3.096537787264044,
|
| 12 |
+
3.031406271457672,
|
| 13 |
+
2.8673954053358597,
|
| 14 |
+
2.7359596317464656,
|
| 15 |
+
2.4514487916773016,
|
| 16 |
+
2.0122493678873234,
|
| 17 |
+
2.0211585906418885,
|
| 18 |
+
2.0931850958954206,
|
| 19 |
+
1.761855153062127,
|
| 20 |
+
1.8128869923678312,
|
| 21 |
+
1.5930419488386673,
|
| 22 |
+
1.5026649328795347,
|
| 23 |
+
1.462189812551845,
|
| 24 |
+
1.2743212932890111,
|
| 25 |
+
1.2935568327253515,
|
| 26 |
+
1.1500763421708888,
|
| 27 |
+
1.14664727286859,
|
| 28 |
+
1.1374793740836058,
|
| 29 |
+
0.9815641880035401,
|
| 30 |
+
0.9445034736936743,
|
| 31 |
+
0.9074165978214958,
|
| 32 |
+
0.9408810431307012,
|
| 33 |
+
0.8156849002296275,
|
| 34 |
+
0.8217221563512629,
|
| 35 |
+
0.7810911265286532,
|
| 36 |
+
0.7809093654155731,
|
| 37 |
+
0.7464128857309168,
|
| 38 |
+
0.7155037416653199,
|
| 39 |
+
0.6639378320087086,
|
| 40 |
+
0.6488462190736424,
|
| 41 |
+
0.6369564533233643,
|
| 42 |
+
0.6154241480610587,
|
| 43 |
+
0.6158316896720366,
|
| 44 |
+
0.5890263763341037,
|
| 45 |
+
0.5694362737915732,
|
| 46 |
+
0.5663888069716367,
|
| 47 |
+
0.547724490815943,
|
| 48 |
+
0.5304673869501461,
|
| 49 |
+
0.5306453683159568,
|
| 50 |
+
0.5292651390487497,
|
| 51 |
+
0.5206340367143805,
|
| 52 |
+
0.525166241960092
|
| 53 |
+
],
|
| 54 |
+
"train_position_loss": [
|
| 55 |
+
984.3100096269087,
|
| 56 |
+
56.93934656489979,
|
| 57 |
+
15.316310223666104,
|
| 58 |
+
8.017440869591454,
|
| 59 |
+
6.128206513144753,
|
| 60 |
+
4.756193041801453,
|
| 61 |
+
5.091845867850564,
|
| 62 |
+
3.3950596007433806,
|
| 63 |
+
3.087064825404774,
|
| 64 |
+
3.022366313500838,
|
| 65 |
+
2.858845823461359,
|
| 66 |
+
2.727849263494665,
|
| 67 |
+
2.443875648758628,
|
| 68 |
+
2.005155376412652,
|
| 69 |
+
2.014330472729423,
|
| 70 |
+
2.0865782743150536,
|
| 71 |
+
1.7556737076152455,
|
| 72 |
+
1.8068639375946738,
|
| 73 |
+
1.5872981082309376,
|
| 74 |
+
1.4971454349431124,
|
| 75 |
+
1.456840877641331,
|
| 76 |
+
1.2692103575576434,
|
| 77 |
+
1.2885674205693332,
|
| 78 |
+
1.1453263840892098,
|
| 79 |
+
1.1420139171860435,
|
| 80 |
+
1.132937333800576,
|
| 81 |
+
0.9772452619942752,
|
| 82 |
+
0.9402903323823756,
|
| 83 |
+
0.9032804375345057,
|
| 84 |
+
0.9368163688616319,
|
| 85 |
+
0.811802110617811,
|
| 86 |
+
0.8178760658610951,
|
| 87 |
+
0.777328804677183,
|
| 88 |
+
0.7771916915069926,
|
| 89 |
+
0.7427612215280532,
|
| 90 |
+
0.7119250953197479,
|
| 91 |
+
0.660411764274944,
|
| 92 |
+
0.645363664356145,
|
| 93 |
+
0.6335091539404609,
|
| 94 |
+
0.6120497462424365,
|
| 95 |
+
0.6124814569950103,
|
| 96 |
+
0.5856986305930397,
|
| 97 |
+
0.566144134239717,
|
| 98 |
+
0.5631175455721942,
|
| 99 |
+
0.5444513786922801,
|
| 100 |
+
0.527230566740036,
|
| 101 |
+
0.527415869994597,
|
| 102 |
+
0.5260429655963724,
|
| 103 |
+
0.5174292992461812,
|
| 104 |
+
0.5219685020771894
|
| 105 |
+
],
|
| 106 |
+
"train_physics_loss": [
|
| 107 |
+
6.89877491755919,
|
| 108 |
+
0.5274312710220164,
|
| 109 |
+
0.24987928718328475,
|
| 110 |
+
0.1695604576983235,
|
| 111 |
+
0.13862182985652577,
|
| 112 |
+
0.12197020880200646,
|
| 113 |
+
0.1164700106463649,
|
| 114 |
+
0.10165808898481456,
|
| 115 |
+
0.09472970481623302,
|
| 116 |
+
0.09039960943839767,
|
| 117 |
+
0.08549566343426704,
|
| 118 |
+
0.0811036845499819,
|
| 119 |
+
0.07573145194487138,
|
| 120 |
+
0.07093993692912838,
|
| 121 |
+
0.06828112229704857,
|
| 122 |
+
0.06606818542561747,
|
| 123 |
+
0.06181450875645334,
|
| 124 |
+
0.06023050509393215,
|
| 125 |
+
0.057438355616547844,
|
| 126 |
+
0.055194985629482704,
|
| 127 |
+
0.05348944115367803,
|
| 128 |
+
0.05110932046716864,
|
| 129 |
+
0.04989412785931067,
|
| 130 |
+
0.047499610382047565,
|
| 131 |
+
0.046333565427498384,
|
| 132 |
+
0.04542038095268336,
|
| 133 |
+
0.04318924254991791,
|
| 134 |
+
0.042131400921128014,
|
| 135 |
+
0.04136156131598082,
|
| 136 |
+
0.04064672789113088,
|
| 137 |
+
0.03882793503051454,
|
| 138 |
+
0.038460931689901785,
|
| 139 |
+
0.037623244320804425,
|
| 140 |
+
0.03717670681124384,
|
| 141 |
+
0.036516643671149554,
|
| 142 |
+
0.03578646706247872,
|
| 143 |
+
0.03526069371199066,
|
| 144 |
+
0.034825517372651535,
|
| 145 |
+
0.03447300194000656,
|
| 146 |
+
0.033744010752574966,
|
| 147 |
+
0.03350233845412731,
|
| 148 |
+
0.033277462812309914,
|
| 149 |
+
0.03292140940373594,
|
| 150 |
+
0.03271263228221373,
|
| 151 |
+
0.03273111722347411,
|
| 152 |
+
0.03236822695894675,
|
| 153 |
+
0.03229496712711724,
|
| 154 |
+
0.032221741043031216,
|
| 155 |
+
0.0320473746989261,
|
| 156 |
+
0.03197741068222306
|
| 157 |
+
],
|
| 158 |
+
"val_loss": [
|
| 159 |
+
153.26430384318033,
|
| 160 |
+
18.892200390497845,
|
| 161 |
+
10.423794229825338,
|
| 162 |
+
4.392075796922048,
|
| 163 |
+
6.797033468882243,
|
| 164 |
+
4.492823789517085,
|
| 165 |
+
3.566956341266632,
|
| 166 |
+
1.7937867641448975,
|
| 167 |
+
3.06752343972524,
|
| 168 |
+
2.7552332331736884,
|
| 169 |
+
3.7816038032372794,
|
| 170 |
+
6.496122757593791,
|
| 171 |
+
1.2621469448010127,
|
| 172 |
+
2.0794981718063354,
|
| 173 |
+
1.3650736634929974,
|
| 174 |
+
0.8943068335453669,
|
| 175 |
+
0.8213488012552261,
|
| 176 |
+
0.8926756928364435,
|
| 177 |
+
0.9191997945308685,
|
| 178 |
+
1.8173842976490657,
|
| 179 |
+
1.474507674574852,
|
| 180 |
+
1.1213604932030041,
|
| 181 |
+
0.8748475636045138,
|
| 182 |
+
0.916106770435969,
|
| 183 |
+
1.3112647905945778,
|
| 184 |
+
0.7422773838043213,
|
| 185 |
+
0.5538644803067049,
|
| 186 |
+
1.1225346277157466,
|
| 187 |
+
0.44924043739835423,
|
| 188 |
+
0.5972975045442581,
|
| 189 |
+
0.57017178957661,
|
| 190 |
+
0.5667421134809653,
|
| 191 |
+
0.6822172378500303,
|
| 192 |
+
0.4893360696732998,
|
| 193 |
+
0.5012122442324957,
|
| 194 |
+
0.5243413771192232,
|
| 195 |
+
0.4362063879768054,
|
| 196 |
+
0.3224947191774845,
|
| 197 |
+
0.37768510977427167,
|
| 198 |
+
0.3410072761277358,
|
| 199 |
+
0.30679150857031345,
|
| 200 |
+
0.377304012576739,
|
| 201 |
+
0.28716951174040634,
|
| 202 |
+
0.28088060207664967,
|
| 203 |
+
0.30061931908130646,
|
| 204 |
+
0.30402864205340546,
|
| 205 |
+
0.2681567470232646,
|
| 206 |
+
0.27820915294190246,
|
| 207 |
+
0.2702796570956707,
|
| 208 |
+
0.27160418157776195
|
| 209 |
+
],
|
| 210 |
+
"val_mae_km": [
|
| 211 |
+
101.554443359375,
|
| 212 |
+
36.752531131108604,
|
| 213 |
+
25.603227694829304,
|
| 214 |
+
20.29919441541036,
|
| 215 |
+
18.206143061319988,
|
| 216 |
+
17.054474115371704,
|
| 217 |
+
15.862754225730896,
|
| 218 |
+
14.423454085985819,
|
| 219 |
+
12.750670313835144,
|
| 220 |
+
14.51679809888204,
|
| 221 |
+
12.564174254735311,
|
| 222 |
+
17.460575222969055,
|
| 223 |
+
11.128435413042704,
|
| 224 |
+
11.643258412679037,
|
| 225 |
+
11.065532724062601,
|
| 226 |
+
9.062310298283895,
|
| 227 |
+
9.068827013174692,
|
| 228 |
+
9.743635714054108,
|
| 229 |
+
9.691622654596964,
|
| 230 |
+
11.47588054339091,
|
| 231 |
+
9.673056920369467,
|
| 232 |
+
9.415653069814047,
|
| 233 |
+
7.796523094177246,
|
| 234 |
+
7.91655969619751,
|
| 235 |
+
8.951678991317749,
|
| 236 |
+
7.278553108374278,
|
| 237 |
+
6.974518994490306,
|
| 238 |
+
8.399892111619314,
|
| 239 |
+
7.136422216892242,
|
| 240 |
+
6.629470308621724,
|
| 241 |
+
7.937336683273315,
|
| 242 |
+
6.252403616905212,
|
| 243 |
+
6.696075896422069,
|
| 244 |
+
6.1597654322783155,
|
| 245 |
+
6.87704195578893,
|
| 246 |
+
7.557182888189952,
|
| 247 |
+
6.164024631182353,
|
| 248 |
+
6.077582021554311,
|
| 249 |
+
6.097978989283244,
|
| 250 |
+
6.394510229428609,
|
| 251 |
+
6.163671771685283,
|
| 252 |
+
6.600313663482666,
|
| 253 |
+
6.370809157689412,
|
| 254 |
+
6.325026532014211,
|
| 255 |
+
5.810727069775264,
|
| 256 |
+
5.929298639297485,
|
| 257 |
+
6.087832132975261,
|
| 258 |
+
6.245916366577148,
|
| 259 |
+
6.049221773942311,
|
| 260 |
+
6.247044622898102
|
| 261 |
+
],
|
| 262 |
+
"lr": [
|
| 263 |
+
9.990232305719945e-05,
|
| 264 |
+
9.960967771506666e-05,
|
| 265 |
+
9.912321891107009e-05,
|
| 266 |
+
9.844486647586723e-05,
|
| 267 |
+
9.757729755661011e-05,
|
| 268 |
+
9.652393605146844e-05,
|
| 269 |
+
9.528893909706797e-05,
|
| 270 |
+
9.387718066217124e-05,
|
| 271 |
+
9.229423231234974e-05,
|
| 272 |
+
9.054634122155989e-05,
|
| 273 |
+
8.864040551740155e-05,
|
| 274 |
+
8.658394705735986e-05,
|
| 275 |
+
8.438508174347008e-05,
|
| 276 |
+
8.205248749256013e-05,
|
| 277 |
+
7.959536998847742e-05,
|
| 278 |
+
7.702342635146032e-05,
|
| 279 |
+
7.434680686803489e-05,
|
| 280 |
+
7.157607493247108e-05,
|
| 281 |
+
6.872216535789155e-05,
|
| 282 |
+
6.579634122155988e-05,
|
| 283 |
+
6.28101494146603e-05,
|
| 284 |
+
5.9775375071993364e-05,
|
| 285 |
+
5.670399506143306e-05,
|
| 286 |
+
5.3608130716701005e-05,
|
| 287 |
+
5.049999999999999e-05,
|
| 288 |
+
4.7391869283298976e-05,
|
| 289 |
+
4.429600493856693e-05,
|
| 290 |
+
4.122462492800662e-05,
|
| 291 |
+
3.8189850585339674e-05,
|
| 292 |
+
3.5203658778440106e-05,
|
| 293 |
+
3.227783464210844e-05,
|
| 294 |
+
2.942392506752889e-05,
|
| 295 |
+
2.6653193131965075e-05,
|
| 296 |
+
2.3976573648539646e-05,
|
| 297 |
+
2.1404630011522576e-05,
|
| 298 |
+
1.8947512507439848e-05,
|
| 299 |
+
1.66149182565299e-05,
|
| 300 |
+
1.4416052942640132e-05,
|
| 301 |
+
1.235959448259843e-05,
|
| 302 |
+
1.04536587784401e-05,
|
| 303 |
+
8.70576768765026e-06,
|
| 304 |
+
7.122819337828748e-06,
|
| 305 |
+
5.711060902932039e-06,
|
| 306 |
+
4.476063948531556e-06,
|
| 307 |
+
3.4227024433898985e-06,
|
| 308 |
+
2.555133524132766e-06,
|
| 309 |
+
1.8767810889299075e-06,
|
| 310 |
+
1.3903222849333503e-06,
|
| 311 |
+
1.0976769428005576e-06,
|
| 312 |
+
1e-06
|
| 313 |
+
]
|
| 314 |
+
}
|