baluk007 commited on
Commit
afe7fa7
·
verified ·
1 Parent(s): fc938e9

Add files using upload-large-folder tool

Browse files
Files changed (5) hide show
  1. README.md +13 -0
  2. best_model.pth +3 -0
  3. config.json +12 -0
  4. final_model.pth +3 -0
  5. 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
+ }