Upload folder using huggingface_hub
Browse filesThis view is limited to 50 files because it contains too many changes.
See raw diff
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/0_client_model_round10.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/0_client_model_round12.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/0_client_model_round15.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/0_client_model_round17.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/0_client_model_round2.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/0_client_model_round20.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/0_client_model_round5.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/0_client_model_round7.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/0_trainer_state.json +392 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/1_client_model_round10.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/1_client_model_round12.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/1_client_model_round15.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/1_client_model_round17.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/1_client_model_round2.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/1_client_model_round20.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/1_client_model_round5.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/1_client_model_round7.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/1_trainer_state.json +392 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/2_client_model_round10.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/2_client_model_round12.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/2_client_model_round15.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/2_client_model_round17.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/2_client_model_round2.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/2_client_model_round20.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/2_client_model_round5.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/2_client_model_round7.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/2_trainer_state.json +392 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/3_client_model_round10.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/3_client_model_round12.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/3_client_model_round15.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/3_client_model_round17.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/3_client_model_round2.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/3_client_model_round20.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/3_client_model_round5.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/3_client_model_round7.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/3_trainer_state.json +392 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/4_client_model_round10.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/4_client_model_round12.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/4_client_model_round15.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/4_client_model_round17.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/4_client_model_round2.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/4_client_model_round20.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/4_client_model_round5.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/4_client_model_round7.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/4_trainer_state.json +392 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/5_client_model_round10.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/5_client_model_round12.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/5_client_model_round15.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/5_client_model_round17.pth +3 -0
- client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/5_client_model_round2.pth +3 -0
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/0_client_model_round10.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:272d152a867ab41665ff4b14bd2664cbe439281aac4ce8b200d7b718307b126b
|
3 |
+
size 360880622
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/0_client_model_round12.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d76b1907e562d7be4b4d8c05be01e4cda31f7db1d20626ea7dfd1af12033ad00
|
3 |
+
size 360880622
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/0_client_model_round15.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ccce5a80eebb1205c033e472b86c672bc6d13cd4c190227bd7e348128e47d581
|
3 |
+
size 360880622
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/0_client_model_round17.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c645bb754c2ce02ce91f973e5140f29977b468544be656eb2bc6d527d6bec808
|
3 |
+
size 360880622
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/0_client_model_round2.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:2c7476432e61646d8e60716a84e633c89b0e944b2cb4ebac0ec4841daee1cde2
|
3 |
+
size 360880106
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/0_client_model_round20.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3c1a84dc752dd6efa6cc783b662d254ed587ca2b9cf76496d9992ff3deb58ee9
|
3 |
+
size 360880622
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/0_client_model_round5.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:10cc54c34e7cd9c8ffe66d0ab7036d02f9ddc8b7839553f6fa9d9fe022abca00
|
3 |
+
size 360880106
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/0_client_model_round7.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:aff43e0efa5d50dda0f87da98d075db9b26ea5d9c7fb5dee923a0c32e15f4a1c
|
3 |
+
size 360880106
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/0_trainer_state.json
ADDED
@@ -0,0 +1,392 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"best_metric": null,
|
3 |
+
"best_model_checkpoint": null,
|
4 |
+
"epoch": 1.0,
|
5 |
+
"eval_steps": 500,
|
6 |
+
"global_step": 100,
|
7 |
+
"is_hyper_param_search": false,
|
8 |
+
"is_local_process_zero": true,
|
9 |
+
"is_world_process_zero": true,
|
10 |
+
"log_history": [
|
11 |
+
{
|
12 |
+
"epoch": 0.02,
|
13 |
+
"grad_norm": 11.468276023864746,
|
14 |
+
"learning_rate": 2e-05,
|
15 |
+
"loss": 1.668,
|
16 |
+
"step": 2
|
17 |
+
},
|
18 |
+
{
|
19 |
+
"epoch": 0.04,
|
20 |
+
"grad_norm": 13.013179779052734,
|
21 |
+
"learning_rate": 2e-05,
|
22 |
+
"loss": 0.6551,
|
23 |
+
"step": 4
|
24 |
+
},
|
25 |
+
{
|
26 |
+
"epoch": 0.06,
|
27 |
+
"grad_norm": 3.899092674255371,
|
28 |
+
"learning_rate": 2e-05,
|
29 |
+
"loss": 1.4227,
|
30 |
+
"step": 6
|
31 |
+
},
|
32 |
+
{
|
33 |
+
"epoch": 0.08,
|
34 |
+
"grad_norm": 10.417702674865723,
|
35 |
+
"learning_rate": 2e-05,
|
36 |
+
"loss": 1.2854,
|
37 |
+
"step": 8
|
38 |
+
},
|
39 |
+
{
|
40 |
+
"epoch": 0.1,
|
41 |
+
"grad_norm": 3.4957451820373535,
|
42 |
+
"learning_rate": 2e-05,
|
43 |
+
"loss": 1.076,
|
44 |
+
"step": 10
|
45 |
+
},
|
46 |
+
{
|
47 |
+
"epoch": 0.12,
|
48 |
+
"grad_norm": 11.509440422058105,
|
49 |
+
"learning_rate": 2e-05,
|
50 |
+
"loss": 0.6792,
|
51 |
+
"step": 12
|
52 |
+
},
|
53 |
+
{
|
54 |
+
"epoch": 0.14,
|
55 |
+
"grad_norm": 5.03311014175415,
|
56 |
+
"learning_rate": 2e-05,
|
57 |
+
"loss": 0.2549,
|
58 |
+
"step": 14
|
59 |
+
},
|
60 |
+
{
|
61 |
+
"epoch": 0.16,
|
62 |
+
"grad_norm": 4.364517688751221,
|
63 |
+
"learning_rate": 2e-05,
|
64 |
+
"loss": 1.4782,
|
65 |
+
"step": 16
|
66 |
+
},
|
67 |
+
{
|
68 |
+
"epoch": 0.18,
|
69 |
+
"grad_norm": 5.390167236328125,
|
70 |
+
"learning_rate": 2e-05,
|
71 |
+
"loss": 0.49,
|
72 |
+
"step": 18
|
73 |
+
},
|
74 |
+
{
|
75 |
+
"epoch": 0.2,
|
76 |
+
"grad_norm": 6.741970062255859,
|
77 |
+
"learning_rate": 2e-05,
|
78 |
+
"loss": 3.6953,
|
79 |
+
"step": 20
|
80 |
+
},
|
81 |
+
{
|
82 |
+
"epoch": 0.22,
|
83 |
+
"grad_norm": 2.631387948989868,
|
84 |
+
"learning_rate": 2e-05,
|
85 |
+
"loss": 0.2798,
|
86 |
+
"step": 22
|
87 |
+
},
|
88 |
+
{
|
89 |
+
"epoch": 0.24,
|
90 |
+
"grad_norm": 7.609371185302734,
|
91 |
+
"learning_rate": 2e-05,
|
92 |
+
"loss": 0.4919,
|
93 |
+
"step": 24
|
94 |
+
},
|
95 |
+
{
|
96 |
+
"epoch": 0.26,
|
97 |
+
"grad_norm": 1.3606047630310059,
|
98 |
+
"learning_rate": 2e-05,
|
99 |
+
"loss": 0.1068,
|
100 |
+
"step": 26
|
101 |
+
},
|
102 |
+
{
|
103 |
+
"epoch": 0.28,
|
104 |
+
"grad_norm": 15.077208518981934,
|
105 |
+
"learning_rate": 2e-05,
|
106 |
+
"loss": 1.6368,
|
107 |
+
"step": 28
|
108 |
+
},
|
109 |
+
{
|
110 |
+
"epoch": 0.3,
|
111 |
+
"grad_norm": 3.3899340629577637,
|
112 |
+
"learning_rate": 2e-05,
|
113 |
+
"loss": 0.274,
|
114 |
+
"step": 30
|
115 |
+
},
|
116 |
+
{
|
117 |
+
"epoch": 0.32,
|
118 |
+
"grad_norm": 5.730544090270996,
|
119 |
+
"learning_rate": 2e-05,
|
120 |
+
"loss": 0.5332,
|
121 |
+
"step": 32
|
122 |
+
},
|
123 |
+
{
|
124 |
+
"epoch": 0.34,
|
125 |
+
"grad_norm": 4.338137149810791,
|
126 |
+
"learning_rate": 2e-05,
|
127 |
+
"loss": 0.5818,
|
128 |
+
"step": 34
|
129 |
+
},
|
130 |
+
{
|
131 |
+
"epoch": 0.36,
|
132 |
+
"grad_norm": 0.47265109419822693,
|
133 |
+
"learning_rate": 2e-05,
|
134 |
+
"loss": 0.3511,
|
135 |
+
"step": 36
|
136 |
+
},
|
137 |
+
{
|
138 |
+
"epoch": 0.38,
|
139 |
+
"grad_norm": 2.583622455596924,
|
140 |
+
"learning_rate": 2e-05,
|
141 |
+
"loss": 0.1571,
|
142 |
+
"step": 38
|
143 |
+
},
|
144 |
+
{
|
145 |
+
"epoch": 0.4,
|
146 |
+
"grad_norm": 5.760534763336182,
|
147 |
+
"learning_rate": 2e-05,
|
148 |
+
"loss": 0.3887,
|
149 |
+
"step": 40
|
150 |
+
},
|
151 |
+
{
|
152 |
+
"epoch": 0.42,
|
153 |
+
"grad_norm": 1.149841547012329,
|
154 |
+
"learning_rate": 2e-05,
|
155 |
+
"loss": 0.4415,
|
156 |
+
"step": 42
|
157 |
+
},
|
158 |
+
{
|
159 |
+
"epoch": 0.44,
|
160 |
+
"grad_norm": 0.4094676077365875,
|
161 |
+
"learning_rate": 2e-05,
|
162 |
+
"loss": 0.858,
|
163 |
+
"step": 44
|
164 |
+
},
|
165 |
+
{
|
166 |
+
"epoch": 0.46,
|
167 |
+
"grad_norm": 4.535027503967285,
|
168 |
+
"learning_rate": 2e-05,
|
169 |
+
"loss": 0.4825,
|
170 |
+
"step": 46
|
171 |
+
},
|
172 |
+
{
|
173 |
+
"epoch": 0.48,
|
174 |
+
"grad_norm": 6.963266372680664,
|
175 |
+
"learning_rate": 2e-05,
|
176 |
+
"loss": 0.542,
|
177 |
+
"step": 48
|
178 |
+
},
|
179 |
+
{
|
180 |
+
"epoch": 0.5,
|
181 |
+
"grad_norm": 0.2709807753562927,
|
182 |
+
"learning_rate": 2e-05,
|
183 |
+
"loss": 0.4804,
|
184 |
+
"step": 50
|
185 |
+
},
|
186 |
+
{
|
187 |
+
"epoch": 0.52,
|
188 |
+
"grad_norm": 1.4240652322769165,
|
189 |
+
"learning_rate": 2e-05,
|
190 |
+
"loss": 0.4452,
|
191 |
+
"step": 52
|
192 |
+
},
|
193 |
+
{
|
194 |
+
"epoch": 0.54,
|
195 |
+
"grad_norm": 3.6048128604888916,
|
196 |
+
"learning_rate": 2e-05,
|
197 |
+
"loss": 0.6762,
|
198 |
+
"step": 54
|
199 |
+
},
|
200 |
+
{
|
201 |
+
"epoch": 0.56,
|
202 |
+
"grad_norm": 2.263965129852295,
|
203 |
+
"learning_rate": 2e-05,
|
204 |
+
"loss": 1.3549,
|
205 |
+
"step": 56
|
206 |
+
},
|
207 |
+
{
|
208 |
+
"epoch": 0.58,
|
209 |
+
"grad_norm": 5.906020164489746,
|
210 |
+
"learning_rate": 2e-05,
|
211 |
+
"loss": 0.7348,
|
212 |
+
"step": 58
|
213 |
+
},
|
214 |
+
{
|
215 |
+
"epoch": 0.6,
|
216 |
+
"grad_norm": 3.632805824279785,
|
217 |
+
"learning_rate": 2e-05,
|
218 |
+
"loss": 0.7929,
|
219 |
+
"step": 60
|
220 |
+
},
|
221 |
+
{
|
222 |
+
"epoch": 0.62,
|
223 |
+
"grad_norm": 2.760554552078247,
|
224 |
+
"learning_rate": 2e-05,
|
225 |
+
"loss": 1.0124,
|
226 |
+
"step": 62
|
227 |
+
},
|
228 |
+
{
|
229 |
+
"epoch": 0.64,
|
230 |
+
"grad_norm": 0.37708696722984314,
|
231 |
+
"learning_rate": 2e-05,
|
232 |
+
"loss": 0.036,
|
233 |
+
"step": 64
|
234 |
+
},
|
235 |
+
{
|
236 |
+
"epoch": 0.66,
|
237 |
+
"grad_norm": 5.953865051269531,
|
238 |
+
"learning_rate": 2e-05,
|
239 |
+
"loss": 1.0636,
|
240 |
+
"step": 66
|
241 |
+
},
|
242 |
+
{
|
243 |
+
"epoch": 0.68,
|
244 |
+
"grad_norm": 0.12408101558685303,
|
245 |
+
"learning_rate": 2e-05,
|
246 |
+
"loss": 0.2447,
|
247 |
+
"step": 68
|
248 |
+
},
|
249 |
+
{
|
250 |
+
"epoch": 0.7,
|
251 |
+
"grad_norm": 1.6159584522247314,
|
252 |
+
"learning_rate": 2e-05,
|
253 |
+
"loss": 1.5753,
|
254 |
+
"step": 70
|
255 |
+
},
|
256 |
+
{
|
257 |
+
"epoch": 0.72,
|
258 |
+
"grad_norm": 2.123723268508911,
|
259 |
+
"learning_rate": 2e-05,
|
260 |
+
"loss": 0.3181,
|
261 |
+
"step": 72
|
262 |
+
},
|
263 |
+
{
|
264 |
+
"epoch": 0.74,
|
265 |
+
"grad_norm": 2.158083200454712,
|
266 |
+
"learning_rate": 2e-05,
|
267 |
+
"loss": 0.255,
|
268 |
+
"step": 74
|
269 |
+
},
|
270 |
+
{
|
271 |
+
"epoch": 0.76,
|
272 |
+
"grad_norm": 3.0079448223114014,
|
273 |
+
"learning_rate": 2e-05,
|
274 |
+
"loss": 1.3879,
|
275 |
+
"step": 76
|
276 |
+
},
|
277 |
+
{
|
278 |
+
"epoch": 0.78,
|
279 |
+
"grad_norm": 3.4162845611572266,
|
280 |
+
"learning_rate": 2e-05,
|
281 |
+
"loss": 1.2661,
|
282 |
+
"step": 78
|
283 |
+
},
|
284 |
+
{
|
285 |
+
"epoch": 0.8,
|
286 |
+
"grad_norm": 0.42022573947906494,
|
287 |
+
"learning_rate": 2e-05,
|
288 |
+
"loss": 0.4624,
|
289 |
+
"step": 80
|
290 |
+
},
|
291 |
+
{
|
292 |
+
"epoch": 0.82,
|
293 |
+
"grad_norm": 2.4718551635742188,
|
294 |
+
"learning_rate": 2e-05,
|
295 |
+
"loss": 2.6513,
|
296 |
+
"step": 82
|
297 |
+
},
|
298 |
+
{
|
299 |
+
"epoch": 0.84,
|
300 |
+
"grad_norm": 3.038799285888672,
|
301 |
+
"learning_rate": 2e-05,
|
302 |
+
"loss": 0.6602,
|
303 |
+
"step": 84
|
304 |
+
},
|
305 |
+
{
|
306 |
+
"epoch": 0.86,
|
307 |
+
"grad_norm": 0.767530620098114,
|
308 |
+
"learning_rate": 2e-05,
|
309 |
+
"loss": 0.2701,
|
310 |
+
"step": 86
|
311 |
+
},
|
312 |
+
{
|
313 |
+
"epoch": 0.88,
|
314 |
+
"grad_norm": 4.633103847503662,
|
315 |
+
"learning_rate": 2e-05,
|
316 |
+
"loss": 0.2863,
|
317 |
+
"step": 88
|
318 |
+
},
|
319 |
+
{
|
320 |
+
"epoch": 0.9,
|
321 |
+
"grad_norm": 0.8834018707275391,
|
322 |
+
"learning_rate": 2e-05,
|
323 |
+
"loss": 0.0962,
|
324 |
+
"step": 90
|
325 |
+
},
|
326 |
+
{
|
327 |
+
"epoch": 0.92,
|
328 |
+
"grad_norm": 0.8029925227165222,
|
329 |
+
"learning_rate": 2e-05,
|
330 |
+
"loss": 0.151,
|
331 |
+
"step": 92
|
332 |
+
},
|
333 |
+
{
|
334 |
+
"epoch": 0.94,
|
335 |
+
"grad_norm": 2.0458297729492188,
|
336 |
+
"learning_rate": 2e-05,
|
337 |
+
"loss": 0.6086,
|
338 |
+
"step": 94
|
339 |
+
},
|
340 |
+
{
|
341 |
+
"epoch": 0.96,
|
342 |
+
"grad_norm": 0.42579904198646545,
|
343 |
+
"learning_rate": 2e-05,
|
344 |
+
"loss": 0.1379,
|
345 |
+
"step": 96
|
346 |
+
},
|
347 |
+
{
|
348 |
+
"epoch": 0.98,
|
349 |
+
"grad_norm": 2.0101704597473145,
|
350 |
+
"learning_rate": 2e-05,
|
351 |
+
"loss": 0.7558,
|
352 |
+
"step": 98
|
353 |
+
},
|
354 |
+
{
|
355 |
+
"epoch": 1.0,
|
356 |
+
"grad_norm": 0.33978980779647827,
|
357 |
+
"learning_rate": 2e-05,
|
358 |
+
"loss": 0.0732,
|
359 |
+
"step": 100
|
360 |
+
},
|
361 |
+
{
|
362 |
+
"epoch": 1.0,
|
363 |
+
"step": 100,
|
364 |
+
"total_flos": 2185662354685952.0,
|
365 |
+
"train_loss": 0.7525365662574768,
|
366 |
+
"train_runtime": 150.9129,
|
367 |
+
"train_samples_per_second": 2.651,
|
368 |
+
"train_steps_per_second": 0.663
|
369 |
+
}
|
370 |
+
],
|
371 |
+
"logging_steps": 2,
|
372 |
+
"max_steps": 100,
|
373 |
+
"num_input_tokens_seen": 0,
|
374 |
+
"num_train_epochs": 1,
|
375 |
+
"save_steps": 500,
|
376 |
+
"stateful_callbacks": {
|
377 |
+
"TrainerControl": {
|
378 |
+
"args": {
|
379 |
+
"should_epoch_stop": false,
|
380 |
+
"should_evaluate": false,
|
381 |
+
"should_log": false,
|
382 |
+
"should_save": false,
|
383 |
+
"should_training_stop": false
|
384 |
+
},
|
385 |
+
"attributes": {}
|
386 |
+
}
|
387 |
+
},
|
388 |
+
"total_flos": 2185662354685952.0,
|
389 |
+
"train_batch_size": 1,
|
390 |
+
"trial_name": null,
|
391 |
+
"trial_params": null
|
392 |
+
}
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/1_client_model_round10.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:aeb4a49b439c28d8d190448191d2491c8f2d9d9381a33bc569753df77b93d107
|
3 |
+
size 360880622
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/1_client_model_round12.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5262adbeb60ccffc1323f73831539fe624c2aed741f137c58f49b119e4fdedae
|
3 |
+
size 360880622
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/1_client_model_round15.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:77cba52695feb8bc4b3ca31a58e5f6683e9fa65f1fb92adb455e7a928e1d7340
|
3 |
+
size 360880622
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/1_client_model_round17.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:58e7cfddf57ab23e40048ac3f369a966ab3599bf543f7d45ca359409a21019f8
|
3 |
+
size 360880622
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/1_client_model_round2.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3f03b72475b164bdca0088703374303e8763eb107dee5edc11c880e977d38854
|
3 |
+
size 360880106
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/1_client_model_round20.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7da1f49acd8f0189d516ddc07a8795e024cda908579a6359e0b0c7660404fca3
|
3 |
+
size 360880622
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/1_client_model_round5.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:fa895286be91de8a6e0f37351e38c8e427030e99aff09b52daf580349c09642a
|
3 |
+
size 360880106
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/1_client_model_round7.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b1bec964f18c50fe82a94a345b89e64ae93b437a2d60349a8bf20f2331b2cfbc
|
3 |
+
size 360880106
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/1_trainer_state.json
ADDED
@@ -0,0 +1,392 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"best_metric": null,
|
3 |
+
"best_model_checkpoint": null,
|
4 |
+
"epoch": 1.0,
|
5 |
+
"eval_steps": 500,
|
6 |
+
"global_step": 100,
|
7 |
+
"is_hyper_param_search": false,
|
8 |
+
"is_local_process_zero": true,
|
9 |
+
"is_world_process_zero": true,
|
10 |
+
"log_history": [
|
11 |
+
{
|
12 |
+
"epoch": 0.02,
|
13 |
+
"grad_norm": 0.9418780207633972,
|
14 |
+
"learning_rate": 2e-05,
|
15 |
+
"loss": 0.3668,
|
16 |
+
"step": 2
|
17 |
+
},
|
18 |
+
{
|
19 |
+
"epoch": 0.04,
|
20 |
+
"grad_norm": 0.7834081053733826,
|
21 |
+
"learning_rate": 2e-05,
|
22 |
+
"loss": 0.2299,
|
23 |
+
"step": 4
|
24 |
+
},
|
25 |
+
{
|
26 |
+
"epoch": 0.06,
|
27 |
+
"grad_norm": 3.570934772491455,
|
28 |
+
"learning_rate": 2e-05,
|
29 |
+
"loss": 0.5088,
|
30 |
+
"step": 6
|
31 |
+
},
|
32 |
+
{
|
33 |
+
"epoch": 0.08,
|
34 |
+
"grad_norm": 2.852219581604004,
|
35 |
+
"learning_rate": 2e-05,
|
36 |
+
"loss": 0.3404,
|
37 |
+
"step": 8
|
38 |
+
},
|
39 |
+
{
|
40 |
+
"epoch": 0.1,
|
41 |
+
"grad_norm": 0.046853017061948776,
|
42 |
+
"learning_rate": 2e-05,
|
43 |
+
"loss": 0.0053,
|
44 |
+
"step": 10
|
45 |
+
},
|
46 |
+
{
|
47 |
+
"epoch": 0.12,
|
48 |
+
"grad_norm": 2.4755373001098633,
|
49 |
+
"learning_rate": 2e-05,
|
50 |
+
"loss": 0.2663,
|
51 |
+
"step": 12
|
52 |
+
},
|
53 |
+
{
|
54 |
+
"epoch": 0.14,
|
55 |
+
"grad_norm": 0.413883239030838,
|
56 |
+
"learning_rate": 2e-05,
|
57 |
+
"loss": 1.0055,
|
58 |
+
"step": 14
|
59 |
+
},
|
60 |
+
{
|
61 |
+
"epoch": 0.16,
|
62 |
+
"grad_norm": 0.07382853329181671,
|
63 |
+
"learning_rate": 2e-05,
|
64 |
+
"loss": 0.3183,
|
65 |
+
"step": 16
|
66 |
+
},
|
67 |
+
{
|
68 |
+
"epoch": 0.18,
|
69 |
+
"grad_norm": 0.10246730595827103,
|
70 |
+
"learning_rate": 2e-05,
|
71 |
+
"loss": 0.0289,
|
72 |
+
"step": 18
|
73 |
+
},
|
74 |
+
{
|
75 |
+
"epoch": 0.2,
|
76 |
+
"grad_norm": 0.03177673742175102,
|
77 |
+
"learning_rate": 2e-05,
|
78 |
+
"loss": 0.0242,
|
79 |
+
"step": 20
|
80 |
+
},
|
81 |
+
{
|
82 |
+
"epoch": 0.22,
|
83 |
+
"grad_norm": 3.1593880653381348,
|
84 |
+
"learning_rate": 2e-05,
|
85 |
+
"loss": 0.4398,
|
86 |
+
"step": 22
|
87 |
+
},
|
88 |
+
{
|
89 |
+
"epoch": 0.24,
|
90 |
+
"grad_norm": 1.9290744066238403,
|
91 |
+
"learning_rate": 2e-05,
|
92 |
+
"loss": 0.0844,
|
93 |
+
"step": 24
|
94 |
+
},
|
95 |
+
{
|
96 |
+
"epoch": 0.26,
|
97 |
+
"grad_norm": 0.05123985931277275,
|
98 |
+
"learning_rate": 2e-05,
|
99 |
+
"loss": 0.0085,
|
100 |
+
"step": 26
|
101 |
+
},
|
102 |
+
{
|
103 |
+
"epoch": 0.28,
|
104 |
+
"grad_norm": 1.8998264074325562,
|
105 |
+
"learning_rate": 2e-05,
|
106 |
+
"loss": 0.527,
|
107 |
+
"step": 28
|
108 |
+
},
|
109 |
+
{
|
110 |
+
"epoch": 0.3,
|
111 |
+
"grad_norm": 0.13683098554611206,
|
112 |
+
"learning_rate": 2e-05,
|
113 |
+
"loss": 0.1605,
|
114 |
+
"step": 30
|
115 |
+
},
|
116 |
+
{
|
117 |
+
"epoch": 0.32,
|
118 |
+
"grad_norm": 0.09497078508138657,
|
119 |
+
"learning_rate": 2e-05,
|
120 |
+
"loss": 0.0061,
|
121 |
+
"step": 32
|
122 |
+
},
|
123 |
+
{
|
124 |
+
"epoch": 0.34,
|
125 |
+
"grad_norm": 0.39149010181427,
|
126 |
+
"learning_rate": 2e-05,
|
127 |
+
"loss": 0.0452,
|
128 |
+
"step": 34
|
129 |
+
},
|
130 |
+
{
|
131 |
+
"epoch": 0.36,
|
132 |
+
"grad_norm": 1.549695372581482,
|
133 |
+
"learning_rate": 2e-05,
|
134 |
+
"loss": 0.1289,
|
135 |
+
"step": 36
|
136 |
+
},
|
137 |
+
{
|
138 |
+
"epoch": 0.38,
|
139 |
+
"grad_norm": 0.633894145488739,
|
140 |
+
"learning_rate": 2e-05,
|
141 |
+
"loss": 0.0119,
|
142 |
+
"step": 38
|
143 |
+
},
|
144 |
+
{
|
145 |
+
"epoch": 0.4,
|
146 |
+
"grad_norm": 0.8946341276168823,
|
147 |
+
"learning_rate": 2e-05,
|
148 |
+
"loss": 0.6402,
|
149 |
+
"step": 40
|
150 |
+
},
|
151 |
+
{
|
152 |
+
"epoch": 0.42,
|
153 |
+
"grad_norm": 7.587536811828613,
|
154 |
+
"learning_rate": 2e-05,
|
155 |
+
"loss": 0.8406,
|
156 |
+
"step": 42
|
157 |
+
},
|
158 |
+
{
|
159 |
+
"epoch": 0.44,
|
160 |
+
"grad_norm": 0.09829995036125183,
|
161 |
+
"learning_rate": 2e-05,
|
162 |
+
"loss": 0.4789,
|
163 |
+
"step": 44
|
164 |
+
},
|
165 |
+
{
|
166 |
+
"epoch": 0.46,
|
167 |
+
"grad_norm": 0.1041155681014061,
|
168 |
+
"learning_rate": 2e-05,
|
169 |
+
"loss": 0.0109,
|
170 |
+
"step": 46
|
171 |
+
},
|
172 |
+
{
|
173 |
+
"epoch": 0.48,
|
174 |
+
"grad_norm": 0.195893332362175,
|
175 |
+
"learning_rate": 2e-05,
|
176 |
+
"loss": 0.0113,
|
177 |
+
"step": 48
|
178 |
+
},
|
179 |
+
{
|
180 |
+
"epoch": 0.5,
|
181 |
+
"grad_norm": 0.05048517510294914,
|
182 |
+
"learning_rate": 2e-05,
|
183 |
+
"loss": 0.0089,
|
184 |
+
"step": 50
|
185 |
+
},
|
186 |
+
{
|
187 |
+
"epoch": 0.52,
|
188 |
+
"grad_norm": 0.13715337216854095,
|
189 |
+
"learning_rate": 2e-05,
|
190 |
+
"loss": 0.0143,
|
191 |
+
"step": 52
|
192 |
+
},
|
193 |
+
{
|
194 |
+
"epoch": 0.54,
|
195 |
+
"grad_norm": 0.015534869395196438,
|
196 |
+
"learning_rate": 2e-05,
|
197 |
+
"loss": 0.0036,
|
198 |
+
"step": 54
|
199 |
+
},
|
200 |
+
{
|
201 |
+
"epoch": 0.56,
|
202 |
+
"grad_norm": 1.8198376893997192,
|
203 |
+
"learning_rate": 2e-05,
|
204 |
+
"loss": 0.1497,
|
205 |
+
"step": 56
|
206 |
+
},
|
207 |
+
{
|
208 |
+
"epoch": 0.58,
|
209 |
+
"grad_norm": 4.195146560668945,
|
210 |
+
"learning_rate": 2e-05,
|
211 |
+
"loss": 0.4006,
|
212 |
+
"step": 58
|
213 |
+
},
|
214 |
+
{
|
215 |
+
"epoch": 0.6,
|
216 |
+
"grad_norm": 0.5402289628982544,
|
217 |
+
"learning_rate": 2e-05,
|
218 |
+
"loss": 0.0427,
|
219 |
+
"step": 60
|
220 |
+
},
|
221 |
+
{
|
222 |
+
"epoch": 0.62,
|
223 |
+
"grad_norm": 0.02903779409825802,
|
224 |
+
"learning_rate": 2e-05,
|
225 |
+
"loss": 0.0064,
|
226 |
+
"step": 62
|
227 |
+
},
|
228 |
+
{
|
229 |
+
"epoch": 0.64,
|
230 |
+
"grad_norm": 0.12855856120586395,
|
231 |
+
"learning_rate": 2e-05,
|
232 |
+
"loss": 0.0144,
|
233 |
+
"step": 64
|
234 |
+
},
|
235 |
+
{
|
236 |
+
"epoch": 0.66,
|
237 |
+
"grad_norm": 3.7946436405181885,
|
238 |
+
"learning_rate": 2e-05,
|
239 |
+
"loss": 0.7804,
|
240 |
+
"step": 66
|
241 |
+
},
|
242 |
+
{
|
243 |
+
"epoch": 0.68,
|
244 |
+
"grad_norm": 0.8190321326255798,
|
245 |
+
"learning_rate": 2e-05,
|
246 |
+
"loss": 0.0701,
|
247 |
+
"step": 68
|
248 |
+
},
|
249 |
+
{
|
250 |
+
"epoch": 0.7,
|
251 |
+
"grad_norm": 0.7014371156692505,
|
252 |
+
"learning_rate": 2e-05,
|
253 |
+
"loss": 0.0854,
|
254 |
+
"step": 70
|
255 |
+
},
|
256 |
+
{
|
257 |
+
"epoch": 0.72,
|
258 |
+
"grad_norm": 0.01210495363920927,
|
259 |
+
"learning_rate": 2e-05,
|
260 |
+
"loss": 0.0023,
|
261 |
+
"step": 72
|
262 |
+
},
|
263 |
+
{
|
264 |
+
"epoch": 0.74,
|
265 |
+
"grad_norm": 0.47593849897384644,
|
266 |
+
"learning_rate": 2e-05,
|
267 |
+
"loss": 0.059,
|
268 |
+
"step": 74
|
269 |
+
},
|
270 |
+
{
|
271 |
+
"epoch": 0.76,
|
272 |
+
"grad_norm": 0.3410400152206421,
|
273 |
+
"learning_rate": 2e-05,
|
274 |
+
"loss": 0.0698,
|
275 |
+
"step": 76
|
276 |
+
},
|
277 |
+
{
|
278 |
+
"epoch": 0.78,
|
279 |
+
"grad_norm": 0.10963509231805801,
|
280 |
+
"learning_rate": 2e-05,
|
281 |
+
"loss": 0.3811,
|
282 |
+
"step": 78
|
283 |
+
},
|
284 |
+
{
|
285 |
+
"epoch": 0.8,
|
286 |
+
"grad_norm": 1.2575267553329468,
|
287 |
+
"learning_rate": 2e-05,
|
288 |
+
"loss": 0.0941,
|
289 |
+
"step": 80
|
290 |
+
},
|
291 |
+
{
|
292 |
+
"epoch": 0.82,
|
293 |
+
"grad_norm": 2.0267863273620605,
|
294 |
+
"learning_rate": 2e-05,
|
295 |
+
"loss": 0.098,
|
296 |
+
"step": 82
|
297 |
+
},
|
298 |
+
{
|
299 |
+
"epoch": 0.84,
|
300 |
+
"grad_norm": 0.1581244319677353,
|
301 |
+
"learning_rate": 2e-05,
|
302 |
+
"loss": 0.0377,
|
303 |
+
"step": 84
|
304 |
+
},
|
305 |
+
{
|
306 |
+
"epoch": 0.86,
|
307 |
+
"grad_norm": 0.07940375804901123,
|
308 |
+
"learning_rate": 2e-05,
|
309 |
+
"loss": 0.0357,
|
310 |
+
"step": 86
|
311 |
+
},
|
312 |
+
{
|
313 |
+
"epoch": 0.88,
|
314 |
+
"grad_norm": 0.023072313517332077,
|
315 |
+
"learning_rate": 2e-05,
|
316 |
+
"loss": 0.0188,
|
317 |
+
"step": 88
|
318 |
+
},
|
319 |
+
{
|
320 |
+
"epoch": 0.9,
|
321 |
+
"grad_norm": 0.7455753087997437,
|
322 |
+
"learning_rate": 2e-05,
|
323 |
+
"loss": 0.0823,
|
324 |
+
"step": 90
|
325 |
+
},
|
326 |
+
{
|
327 |
+
"epoch": 0.92,
|
328 |
+
"grad_norm": 0.048216842114925385,
|
329 |
+
"learning_rate": 2e-05,
|
330 |
+
"loss": 0.1018,
|
331 |
+
"step": 92
|
332 |
+
},
|
333 |
+
{
|
334 |
+
"epoch": 0.94,
|
335 |
+
"grad_norm": 2.0343430042266846,
|
336 |
+
"learning_rate": 2e-05,
|
337 |
+
"loss": 0.176,
|
338 |
+
"step": 94
|
339 |
+
},
|
340 |
+
{
|
341 |
+
"epoch": 0.96,
|
342 |
+
"grad_norm": 0.007752849720418453,
|
343 |
+
"learning_rate": 2e-05,
|
344 |
+
"loss": 0.0009,
|
345 |
+
"step": 96
|
346 |
+
},
|
347 |
+
{
|
348 |
+
"epoch": 0.98,
|
349 |
+
"grad_norm": 0.03207560256123543,
|
350 |
+
"learning_rate": 2e-05,
|
351 |
+
"loss": 0.003,
|
352 |
+
"step": 98
|
353 |
+
},
|
354 |
+
{
|
355 |
+
"epoch": 1.0,
|
356 |
+
"grad_norm": 0.009924360550940037,
|
357 |
+
"learning_rate": 2e-05,
|
358 |
+
"loss": 0.0046,
|
359 |
+
"step": 100
|
360 |
+
},
|
361 |
+
{
|
362 |
+
"epoch": 1.0,
|
363 |
+
"step": 100,
|
364 |
+
"total_flos": 2203100823683072.0,
|
365 |
+
"train_loss": 0.18460362195968627,
|
366 |
+
"train_runtime": 149.7722,
|
367 |
+
"train_samples_per_second": 2.671,
|
368 |
+
"train_steps_per_second": 0.668
|
369 |
+
}
|
370 |
+
],
|
371 |
+
"logging_steps": 2,
|
372 |
+
"max_steps": 100,
|
373 |
+
"num_input_tokens_seen": 0,
|
374 |
+
"num_train_epochs": 1,
|
375 |
+
"save_steps": 500,
|
376 |
+
"stateful_callbacks": {
|
377 |
+
"TrainerControl": {
|
378 |
+
"args": {
|
379 |
+
"should_epoch_stop": false,
|
380 |
+
"should_evaluate": false,
|
381 |
+
"should_log": false,
|
382 |
+
"should_save": false,
|
383 |
+
"should_training_stop": false
|
384 |
+
},
|
385 |
+
"attributes": {}
|
386 |
+
}
|
387 |
+
},
|
388 |
+
"total_flos": 2203100823683072.0,
|
389 |
+
"train_batch_size": 1,
|
390 |
+
"trial_name": null,
|
391 |
+
"trial_params": null
|
392 |
+
}
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/2_client_model_round10.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a019c92267ae5e335d1fc8ff84c7bc54d54e96aba66854e764ac80751b6f099d
|
3 |
+
size 778341886
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/2_client_model_round12.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6242ea4cc9a3597dc8837aaecf93606e3398073b2c82117d2a6a0ce78f98e8f9
|
3 |
+
size 778341886
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/2_client_model_round15.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7f55720eb65ef087919dc1066498f361c45fc3eddb2c5e4e5dabc94a09b7e9d9
|
3 |
+
size 778341886
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/2_client_model_round17.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:4a41aa0790261c9acffea762697a87338db5625da24eb92c86311cc1d1f6e6e3
|
3 |
+
size 778341886
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/2_client_model_round2.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b3db03128dae4ecfbb360db76f763c5031af8aca8d8151ea474128fe2c4143c3
|
3 |
+
size 778341034
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/2_client_model_round20.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c0989773b3a23802284d1201822033ed34046005735930e560fed0be9b06a3f4
|
3 |
+
size 778341886
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/2_client_model_round5.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:da6040888c80e53df52e617b46f5db4f3d546eacccbf52b2d80a692324698721
|
3 |
+
size 778341034
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/2_client_model_round7.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f29dd2aaaa2d2887d9ddd03c9ec0218e80c11ab8e26af8c70b5e443345dac2df
|
3 |
+
size 778341034
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/2_trainer_state.json
ADDED
@@ -0,0 +1,392 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"best_metric": null,
|
3 |
+
"best_model_checkpoint": null,
|
4 |
+
"epoch": 1.0,
|
5 |
+
"eval_steps": 500,
|
6 |
+
"global_step": 100,
|
7 |
+
"is_hyper_param_search": false,
|
8 |
+
"is_local_process_zero": true,
|
9 |
+
"is_world_process_zero": true,
|
10 |
+
"log_history": [
|
11 |
+
{
|
12 |
+
"epoch": 0.02,
|
13 |
+
"grad_norm": 2.7772328853607178,
|
14 |
+
"learning_rate": 2e-05,
|
15 |
+
"loss": 1.0009,
|
16 |
+
"step": 2
|
17 |
+
},
|
18 |
+
{
|
19 |
+
"epoch": 0.04,
|
20 |
+
"grad_norm": 2.870094060897827,
|
21 |
+
"learning_rate": 2e-05,
|
22 |
+
"loss": 0.7007,
|
23 |
+
"step": 4
|
24 |
+
},
|
25 |
+
{
|
26 |
+
"epoch": 0.06,
|
27 |
+
"grad_norm": 0.33457279205322266,
|
28 |
+
"learning_rate": 2e-05,
|
29 |
+
"loss": 0.0473,
|
30 |
+
"step": 6
|
31 |
+
},
|
32 |
+
{
|
33 |
+
"epoch": 0.08,
|
34 |
+
"grad_norm": 1.4138612747192383,
|
35 |
+
"learning_rate": 2e-05,
|
36 |
+
"loss": 0.2522,
|
37 |
+
"step": 8
|
38 |
+
},
|
39 |
+
{
|
40 |
+
"epoch": 0.1,
|
41 |
+
"grad_norm": 1.5410336256027222,
|
42 |
+
"learning_rate": 2e-05,
|
43 |
+
"loss": 0.2067,
|
44 |
+
"step": 10
|
45 |
+
},
|
46 |
+
{
|
47 |
+
"epoch": 0.12,
|
48 |
+
"grad_norm": 1.8038370609283447,
|
49 |
+
"learning_rate": 2e-05,
|
50 |
+
"loss": 0.6918,
|
51 |
+
"step": 12
|
52 |
+
},
|
53 |
+
{
|
54 |
+
"epoch": 0.14,
|
55 |
+
"grad_norm": 6.221210479736328,
|
56 |
+
"learning_rate": 2e-05,
|
57 |
+
"loss": 1.1887,
|
58 |
+
"step": 14
|
59 |
+
},
|
60 |
+
{
|
61 |
+
"epoch": 0.16,
|
62 |
+
"grad_norm": 2.4460601806640625,
|
63 |
+
"learning_rate": 2e-05,
|
64 |
+
"loss": 0.7211,
|
65 |
+
"step": 16
|
66 |
+
},
|
67 |
+
{
|
68 |
+
"epoch": 0.18,
|
69 |
+
"grad_norm": 0.6152211427688599,
|
70 |
+
"learning_rate": 2e-05,
|
71 |
+
"loss": 0.3936,
|
72 |
+
"step": 18
|
73 |
+
},
|
74 |
+
{
|
75 |
+
"epoch": 0.2,
|
76 |
+
"grad_norm": 4.758711338043213,
|
77 |
+
"learning_rate": 2e-05,
|
78 |
+
"loss": 1.348,
|
79 |
+
"step": 20
|
80 |
+
},
|
81 |
+
{
|
82 |
+
"epoch": 0.22,
|
83 |
+
"grad_norm": 2.9407858848571777,
|
84 |
+
"learning_rate": 2e-05,
|
85 |
+
"loss": 0.4284,
|
86 |
+
"step": 22
|
87 |
+
},
|
88 |
+
{
|
89 |
+
"epoch": 0.24,
|
90 |
+
"grad_norm": 2.8231184482574463,
|
91 |
+
"learning_rate": 2e-05,
|
92 |
+
"loss": 0.289,
|
93 |
+
"step": 24
|
94 |
+
},
|
95 |
+
{
|
96 |
+
"epoch": 0.26,
|
97 |
+
"grad_norm": 1.1073336601257324,
|
98 |
+
"learning_rate": 2e-05,
|
99 |
+
"loss": 0.1379,
|
100 |
+
"step": 26
|
101 |
+
},
|
102 |
+
{
|
103 |
+
"epoch": 0.28,
|
104 |
+
"grad_norm": 3.671870708465576,
|
105 |
+
"learning_rate": 2e-05,
|
106 |
+
"loss": 1.0666,
|
107 |
+
"step": 28
|
108 |
+
},
|
109 |
+
{
|
110 |
+
"epoch": 0.3,
|
111 |
+
"grad_norm": 2.4478886127471924,
|
112 |
+
"learning_rate": 2e-05,
|
113 |
+
"loss": 1.5288,
|
114 |
+
"step": 30
|
115 |
+
},
|
116 |
+
{
|
117 |
+
"epoch": 0.32,
|
118 |
+
"grad_norm": 2.0650575160980225,
|
119 |
+
"learning_rate": 2e-05,
|
120 |
+
"loss": 0.6413,
|
121 |
+
"step": 32
|
122 |
+
},
|
123 |
+
{
|
124 |
+
"epoch": 0.34,
|
125 |
+
"grad_norm": 4.9534478187561035,
|
126 |
+
"learning_rate": 2e-05,
|
127 |
+
"loss": 1.518,
|
128 |
+
"step": 34
|
129 |
+
},
|
130 |
+
{
|
131 |
+
"epoch": 0.36,
|
132 |
+
"grad_norm": 0.6580589413642883,
|
133 |
+
"learning_rate": 2e-05,
|
134 |
+
"loss": 0.0963,
|
135 |
+
"step": 36
|
136 |
+
},
|
137 |
+
{
|
138 |
+
"epoch": 0.38,
|
139 |
+
"grad_norm": 3.114241600036621,
|
140 |
+
"learning_rate": 2e-05,
|
141 |
+
"loss": 0.5737,
|
142 |
+
"step": 38
|
143 |
+
},
|
144 |
+
{
|
145 |
+
"epoch": 0.4,
|
146 |
+
"grad_norm": 0.5105870366096497,
|
147 |
+
"learning_rate": 2e-05,
|
148 |
+
"loss": 0.5189,
|
149 |
+
"step": 40
|
150 |
+
},
|
151 |
+
{
|
152 |
+
"epoch": 0.42,
|
153 |
+
"grad_norm": 0.34910497069358826,
|
154 |
+
"learning_rate": 2e-05,
|
155 |
+
"loss": 0.0478,
|
156 |
+
"step": 42
|
157 |
+
},
|
158 |
+
{
|
159 |
+
"epoch": 0.44,
|
160 |
+
"grad_norm": 2.300654411315918,
|
161 |
+
"learning_rate": 2e-05,
|
162 |
+
"loss": 0.3245,
|
163 |
+
"step": 44
|
164 |
+
},
|
165 |
+
{
|
166 |
+
"epoch": 0.46,
|
167 |
+
"grad_norm": 1.8640389442443848,
|
168 |
+
"learning_rate": 2e-05,
|
169 |
+
"loss": 0.2942,
|
170 |
+
"step": 46
|
171 |
+
},
|
172 |
+
{
|
173 |
+
"epoch": 0.48,
|
174 |
+
"grad_norm": 0.876298725605011,
|
175 |
+
"learning_rate": 2e-05,
|
176 |
+
"loss": 0.3075,
|
177 |
+
"step": 48
|
178 |
+
},
|
179 |
+
{
|
180 |
+
"epoch": 0.5,
|
181 |
+
"grad_norm": 0.08325261622667313,
|
182 |
+
"learning_rate": 2e-05,
|
183 |
+
"loss": 0.2153,
|
184 |
+
"step": 50
|
185 |
+
},
|
186 |
+
{
|
187 |
+
"epoch": 0.52,
|
188 |
+
"grad_norm": 0.11174854636192322,
|
189 |
+
"learning_rate": 2e-05,
|
190 |
+
"loss": 0.4341,
|
191 |
+
"step": 52
|
192 |
+
},
|
193 |
+
{
|
194 |
+
"epoch": 0.54,
|
195 |
+
"grad_norm": 1.753587245941162,
|
196 |
+
"learning_rate": 2e-05,
|
197 |
+
"loss": 0.8374,
|
198 |
+
"step": 54
|
199 |
+
},
|
200 |
+
{
|
201 |
+
"epoch": 0.56,
|
202 |
+
"grad_norm": 3.9684956073760986,
|
203 |
+
"learning_rate": 2e-05,
|
204 |
+
"loss": 1.4496,
|
205 |
+
"step": 56
|
206 |
+
},
|
207 |
+
{
|
208 |
+
"epoch": 0.58,
|
209 |
+
"grad_norm": 2.8988077640533447,
|
210 |
+
"learning_rate": 2e-05,
|
211 |
+
"loss": 1.2955,
|
212 |
+
"step": 58
|
213 |
+
},
|
214 |
+
{
|
215 |
+
"epoch": 0.6,
|
216 |
+
"grad_norm": 1.069868564605713,
|
217 |
+
"learning_rate": 2e-05,
|
218 |
+
"loss": 0.4547,
|
219 |
+
"step": 60
|
220 |
+
},
|
221 |
+
{
|
222 |
+
"epoch": 0.62,
|
223 |
+
"grad_norm": 0.5531668663024902,
|
224 |
+
"learning_rate": 2e-05,
|
225 |
+
"loss": 0.1617,
|
226 |
+
"step": 62
|
227 |
+
},
|
228 |
+
{
|
229 |
+
"epoch": 0.64,
|
230 |
+
"grad_norm": 1.0127068758010864,
|
231 |
+
"learning_rate": 2e-05,
|
232 |
+
"loss": 0.5601,
|
233 |
+
"step": 64
|
234 |
+
},
|
235 |
+
{
|
236 |
+
"epoch": 0.66,
|
237 |
+
"grad_norm": 0.7856873273849487,
|
238 |
+
"learning_rate": 2e-05,
|
239 |
+
"loss": 0.3769,
|
240 |
+
"step": 66
|
241 |
+
},
|
242 |
+
{
|
243 |
+
"epoch": 0.68,
|
244 |
+
"grad_norm": 1.4297077655792236,
|
245 |
+
"learning_rate": 2e-05,
|
246 |
+
"loss": 0.5413,
|
247 |
+
"step": 68
|
248 |
+
},
|
249 |
+
{
|
250 |
+
"epoch": 0.7,
|
251 |
+
"grad_norm": 0.4274978041648865,
|
252 |
+
"learning_rate": 2e-05,
|
253 |
+
"loss": 0.2912,
|
254 |
+
"step": 70
|
255 |
+
},
|
256 |
+
{
|
257 |
+
"epoch": 0.72,
|
258 |
+
"grad_norm": 1.011972427368164,
|
259 |
+
"learning_rate": 2e-05,
|
260 |
+
"loss": 0.6817,
|
261 |
+
"step": 72
|
262 |
+
},
|
263 |
+
{
|
264 |
+
"epoch": 0.74,
|
265 |
+
"grad_norm": 1.2338765859603882,
|
266 |
+
"learning_rate": 2e-05,
|
267 |
+
"loss": 0.5988,
|
268 |
+
"step": 74
|
269 |
+
},
|
270 |
+
{
|
271 |
+
"epoch": 0.76,
|
272 |
+
"grad_norm": 0.7795504331588745,
|
273 |
+
"learning_rate": 2e-05,
|
274 |
+
"loss": 0.4102,
|
275 |
+
"step": 76
|
276 |
+
},
|
277 |
+
{
|
278 |
+
"epoch": 0.78,
|
279 |
+
"grad_norm": 0.6567718982696533,
|
280 |
+
"learning_rate": 2e-05,
|
281 |
+
"loss": 0.4535,
|
282 |
+
"step": 78
|
283 |
+
},
|
284 |
+
{
|
285 |
+
"epoch": 0.8,
|
286 |
+
"grad_norm": 1.0618244409561157,
|
287 |
+
"learning_rate": 2e-05,
|
288 |
+
"loss": 1.3002,
|
289 |
+
"step": 80
|
290 |
+
},
|
291 |
+
{
|
292 |
+
"epoch": 0.82,
|
293 |
+
"grad_norm": 0.8827853798866272,
|
294 |
+
"learning_rate": 2e-05,
|
295 |
+
"loss": 0.7112,
|
296 |
+
"step": 82
|
297 |
+
},
|
298 |
+
{
|
299 |
+
"epoch": 0.84,
|
300 |
+
"grad_norm": 0.621549129486084,
|
301 |
+
"learning_rate": 2e-05,
|
302 |
+
"loss": 0.414,
|
303 |
+
"step": 84
|
304 |
+
},
|
305 |
+
{
|
306 |
+
"epoch": 0.86,
|
307 |
+
"grad_norm": 3.6500234603881836,
|
308 |
+
"learning_rate": 2e-05,
|
309 |
+
"loss": 1.0575,
|
310 |
+
"step": 86
|
311 |
+
},
|
312 |
+
{
|
313 |
+
"epoch": 0.88,
|
314 |
+
"grad_norm": 0.41276150941848755,
|
315 |
+
"learning_rate": 2e-05,
|
316 |
+
"loss": 0.1767,
|
317 |
+
"step": 88
|
318 |
+
},
|
319 |
+
{
|
320 |
+
"epoch": 0.9,
|
321 |
+
"grad_norm": 2.3679842948913574,
|
322 |
+
"learning_rate": 2e-05,
|
323 |
+
"loss": 0.2918,
|
324 |
+
"step": 90
|
325 |
+
},
|
326 |
+
{
|
327 |
+
"epoch": 0.92,
|
328 |
+
"grad_norm": 1.2793606519699097,
|
329 |
+
"learning_rate": 2e-05,
|
330 |
+
"loss": 0.8175,
|
331 |
+
"step": 92
|
332 |
+
},
|
333 |
+
{
|
334 |
+
"epoch": 0.94,
|
335 |
+
"grad_norm": 3.083470106124878,
|
336 |
+
"learning_rate": 2e-05,
|
337 |
+
"loss": 0.6887,
|
338 |
+
"step": 94
|
339 |
+
},
|
340 |
+
{
|
341 |
+
"epoch": 0.96,
|
342 |
+
"grad_norm": 0.6317909955978394,
|
343 |
+
"learning_rate": 2e-05,
|
344 |
+
"loss": 0.6057,
|
345 |
+
"step": 96
|
346 |
+
},
|
347 |
+
{
|
348 |
+
"epoch": 0.98,
|
349 |
+
"grad_norm": 1.9122306108474731,
|
350 |
+
"learning_rate": 2e-05,
|
351 |
+
"loss": 0.6412,
|
352 |
+
"step": 98
|
353 |
+
},
|
354 |
+
{
|
355 |
+
"epoch": 1.0,
|
356 |
+
"grad_norm": 3.5604162216186523,
|
357 |
+
"learning_rate": 2e-05,
|
358 |
+
"loss": 0.7647,
|
359 |
+
"step": 100
|
360 |
+
},
|
361 |
+
{
|
362 |
+
"epoch": 1.0,
|
363 |
+
"step": 100,
|
364 |
+
"total_flos": 5200945432494080.0,
|
365 |
+
"train_loss": 0.6110978984832763,
|
366 |
+
"train_runtime": 254.1782,
|
367 |
+
"train_samples_per_second": 1.574,
|
368 |
+
"train_steps_per_second": 0.393
|
369 |
+
}
|
370 |
+
],
|
371 |
+
"logging_steps": 2,
|
372 |
+
"max_steps": 100,
|
373 |
+
"num_input_tokens_seen": 0,
|
374 |
+
"num_train_epochs": 1,
|
375 |
+
"save_steps": 500,
|
376 |
+
"stateful_callbacks": {
|
377 |
+
"TrainerControl": {
|
378 |
+
"args": {
|
379 |
+
"should_epoch_stop": false,
|
380 |
+
"should_evaluate": false,
|
381 |
+
"should_log": false,
|
382 |
+
"should_save": false,
|
383 |
+
"should_training_stop": false
|
384 |
+
},
|
385 |
+
"attributes": {}
|
386 |
+
}
|
387 |
+
},
|
388 |
+
"total_flos": 5200945432494080.0,
|
389 |
+
"train_batch_size": 1,
|
390 |
+
"trial_name": null,
|
391 |
+
"trial_params": null
|
392 |
+
}
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/3_client_model_round10.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a09dc85a329ffdfcb332f4a210c4251b86b95e5d6dc5312a63f5a569f098b29a
|
3 |
+
size 360880622
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/3_client_model_round12.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:67938ab6eeb915d73a553f80721eb8e49fa1ec3af8c756286e61bb00fba2b40c
|
3 |
+
size 360880622
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/3_client_model_round15.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:fa9d648dd6669c20a5b07d0a810eabe043d67ec4ebc16d7bb2a7ca44fdea0a05
|
3 |
+
size 360880622
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/3_client_model_round17.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:412e5c6b8d09bafb8dfffb4e470a1accfa6670ee17d7894b972774a99b001990
|
3 |
+
size 360880622
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/3_client_model_round2.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e372c275a65c800dee474024b429c87baefb6e713548d7392a255d0299246ca5
|
3 |
+
size 360880106
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/3_client_model_round20.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0353c976d6a405c39434e2fbfead24f4396b1945cd0c74ea8e3025c308dd6b53
|
3 |
+
size 360880622
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/3_client_model_round5.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b44cf92c0b4f0c4f48eeb4c53de6cf30bce271873da0250b2baee6543caf7605
|
3 |
+
size 360880106
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/3_client_model_round7.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:33b9cb7fdf91a5d699f2b4ec2b6cd3c5fe3e4859c5e09da3ab2c6d489b753783
|
3 |
+
size 360880106
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/3_trainer_state.json
ADDED
@@ -0,0 +1,392 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"best_metric": null,
|
3 |
+
"best_model_checkpoint": null,
|
4 |
+
"epoch": 1.0,
|
5 |
+
"eval_steps": 500,
|
6 |
+
"global_step": 100,
|
7 |
+
"is_hyper_param_search": false,
|
8 |
+
"is_local_process_zero": true,
|
9 |
+
"is_world_process_zero": true,
|
10 |
+
"log_history": [
|
11 |
+
{
|
12 |
+
"epoch": 0.02,
|
13 |
+
"grad_norm": 7.792142391204834,
|
14 |
+
"learning_rate": 2e-05,
|
15 |
+
"loss": 1.2916,
|
16 |
+
"step": 2
|
17 |
+
},
|
18 |
+
{
|
19 |
+
"epoch": 0.04,
|
20 |
+
"grad_norm": 6.023014545440674,
|
21 |
+
"learning_rate": 2e-05,
|
22 |
+
"loss": 3.5714,
|
23 |
+
"step": 4
|
24 |
+
},
|
25 |
+
{
|
26 |
+
"epoch": 0.06,
|
27 |
+
"grad_norm": 6.57926082611084,
|
28 |
+
"learning_rate": 2e-05,
|
29 |
+
"loss": 1.5864,
|
30 |
+
"step": 6
|
31 |
+
},
|
32 |
+
{
|
33 |
+
"epoch": 0.08,
|
34 |
+
"grad_norm": 10.789443016052246,
|
35 |
+
"learning_rate": 2e-05,
|
36 |
+
"loss": 2.7961,
|
37 |
+
"step": 8
|
38 |
+
},
|
39 |
+
{
|
40 |
+
"epoch": 0.1,
|
41 |
+
"grad_norm": 3.5799789428710938,
|
42 |
+
"learning_rate": 2e-05,
|
43 |
+
"loss": 0.5679,
|
44 |
+
"step": 10
|
45 |
+
},
|
46 |
+
{
|
47 |
+
"epoch": 0.12,
|
48 |
+
"grad_norm": 4.088947772979736,
|
49 |
+
"learning_rate": 2e-05,
|
50 |
+
"loss": 1.5752,
|
51 |
+
"step": 12
|
52 |
+
},
|
53 |
+
{
|
54 |
+
"epoch": 0.14,
|
55 |
+
"grad_norm": 5.1460089683532715,
|
56 |
+
"learning_rate": 2e-05,
|
57 |
+
"loss": 0.9169,
|
58 |
+
"step": 14
|
59 |
+
},
|
60 |
+
{
|
61 |
+
"epoch": 0.16,
|
62 |
+
"grad_norm": 3.7856991291046143,
|
63 |
+
"learning_rate": 2e-05,
|
64 |
+
"loss": 0.8518,
|
65 |
+
"step": 16
|
66 |
+
},
|
67 |
+
{
|
68 |
+
"epoch": 0.18,
|
69 |
+
"grad_norm": 5.619609832763672,
|
70 |
+
"learning_rate": 2e-05,
|
71 |
+
"loss": 1.3801,
|
72 |
+
"step": 18
|
73 |
+
},
|
74 |
+
{
|
75 |
+
"epoch": 0.2,
|
76 |
+
"grad_norm": 2.50746488571167,
|
77 |
+
"learning_rate": 2e-05,
|
78 |
+
"loss": 1.2587,
|
79 |
+
"step": 20
|
80 |
+
},
|
81 |
+
{
|
82 |
+
"epoch": 0.22,
|
83 |
+
"grad_norm": 8.305500030517578,
|
84 |
+
"learning_rate": 2e-05,
|
85 |
+
"loss": 1.0734,
|
86 |
+
"step": 22
|
87 |
+
},
|
88 |
+
{
|
89 |
+
"epoch": 0.24,
|
90 |
+
"grad_norm": 5.19241189956665,
|
91 |
+
"learning_rate": 2e-05,
|
92 |
+
"loss": 0.7836,
|
93 |
+
"step": 24
|
94 |
+
},
|
95 |
+
{
|
96 |
+
"epoch": 0.26,
|
97 |
+
"grad_norm": 3.3395862579345703,
|
98 |
+
"learning_rate": 2e-05,
|
99 |
+
"loss": 0.6475,
|
100 |
+
"step": 26
|
101 |
+
},
|
102 |
+
{
|
103 |
+
"epoch": 0.28,
|
104 |
+
"grad_norm": 7.040569305419922,
|
105 |
+
"learning_rate": 2e-05,
|
106 |
+
"loss": 1.2521,
|
107 |
+
"step": 28
|
108 |
+
},
|
109 |
+
{
|
110 |
+
"epoch": 0.3,
|
111 |
+
"grad_norm": 4.001715660095215,
|
112 |
+
"learning_rate": 2e-05,
|
113 |
+
"loss": 1.1543,
|
114 |
+
"step": 30
|
115 |
+
},
|
116 |
+
{
|
117 |
+
"epoch": 0.32,
|
118 |
+
"grad_norm": 8.984875679016113,
|
119 |
+
"learning_rate": 2e-05,
|
120 |
+
"loss": 2.0586,
|
121 |
+
"step": 32
|
122 |
+
},
|
123 |
+
{
|
124 |
+
"epoch": 0.34,
|
125 |
+
"grad_norm": 4.349306583404541,
|
126 |
+
"learning_rate": 2e-05,
|
127 |
+
"loss": 1.0413,
|
128 |
+
"step": 34
|
129 |
+
},
|
130 |
+
{
|
131 |
+
"epoch": 0.36,
|
132 |
+
"grad_norm": 5.427790641784668,
|
133 |
+
"learning_rate": 2e-05,
|
134 |
+
"loss": 1.6563,
|
135 |
+
"step": 36
|
136 |
+
},
|
137 |
+
{
|
138 |
+
"epoch": 0.38,
|
139 |
+
"grad_norm": 1.6137840747833252,
|
140 |
+
"learning_rate": 2e-05,
|
141 |
+
"loss": 1.1592,
|
142 |
+
"step": 38
|
143 |
+
},
|
144 |
+
{
|
145 |
+
"epoch": 0.4,
|
146 |
+
"grad_norm": 2.16748309135437,
|
147 |
+
"learning_rate": 2e-05,
|
148 |
+
"loss": 0.5156,
|
149 |
+
"step": 40
|
150 |
+
},
|
151 |
+
{
|
152 |
+
"epoch": 0.42,
|
153 |
+
"grad_norm": 8.764156341552734,
|
154 |
+
"learning_rate": 2e-05,
|
155 |
+
"loss": 2.4003,
|
156 |
+
"step": 42
|
157 |
+
},
|
158 |
+
{
|
159 |
+
"epoch": 0.44,
|
160 |
+
"grad_norm": 9.373359680175781,
|
161 |
+
"learning_rate": 2e-05,
|
162 |
+
"loss": 1.6908,
|
163 |
+
"step": 44
|
164 |
+
},
|
165 |
+
{
|
166 |
+
"epoch": 0.46,
|
167 |
+
"grad_norm": 2.3396241664886475,
|
168 |
+
"learning_rate": 2e-05,
|
169 |
+
"loss": 0.7827,
|
170 |
+
"step": 46
|
171 |
+
},
|
172 |
+
{
|
173 |
+
"epoch": 0.48,
|
174 |
+
"grad_norm": 2.254624128341675,
|
175 |
+
"learning_rate": 2e-05,
|
176 |
+
"loss": 0.5496,
|
177 |
+
"step": 48
|
178 |
+
},
|
179 |
+
{
|
180 |
+
"epoch": 0.5,
|
181 |
+
"grad_norm": 4.849947929382324,
|
182 |
+
"learning_rate": 2e-05,
|
183 |
+
"loss": 1.1958,
|
184 |
+
"step": 50
|
185 |
+
},
|
186 |
+
{
|
187 |
+
"epoch": 0.52,
|
188 |
+
"grad_norm": 2.5219838619232178,
|
189 |
+
"learning_rate": 2e-05,
|
190 |
+
"loss": 0.9757,
|
191 |
+
"step": 52
|
192 |
+
},
|
193 |
+
{
|
194 |
+
"epoch": 0.54,
|
195 |
+
"grad_norm": 2.996366262435913,
|
196 |
+
"learning_rate": 2e-05,
|
197 |
+
"loss": 0.8615,
|
198 |
+
"step": 54
|
199 |
+
},
|
200 |
+
{
|
201 |
+
"epoch": 0.56,
|
202 |
+
"grad_norm": 3.0587832927703857,
|
203 |
+
"learning_rate": 2e-05,
|
204 |
+
"loss": 0.9333,
|
205 |
+
"step": 56
|
206 |
+
},
|
207 |
+
{
|
208 |
+
"epoch": 0.58,
|
209 |
+
"grad_norm": 8.328261375427246,
|
210 |
+
"learning_rate": 2e-05,
|
211 |
+
"loss": 2.0392,
|
212 |
+
"step": 58
|
213 |
+
},
|
214 |
+
{
|
215 |
+
"epoch": 0.6,
|
216 |
+
"grad_norm": 1.4318846464157104,
|
217 |
+
"learning_rate": 2e-05,
|
218 |
+
"loss": 1.6461,
|
219 |
+
"step": 60
|
220 |
+
},
|
221 |
+
{
|
222 |
+
"epoch": 0.62,
|
223 |
+
"grad_norm": 1.621255874633789,
|
224 |
+
"learning_rate": 2e-05,
|
225 |
+
"loss": 0.4382,
|
226 |
+
"step": 62
|
227 |
+
},
|
228 |
+
{
|
229 |
+
"epoch": 0.64,
|
230 |
+
"grad_norm": 5.432891845703125,
|
231 |
+
"learning_rate": 2e-05,
|
232 |
+
"loss": 1.6557,
|
233 |
+
"step": 64
|
234 |
+
},
|
235 |
+
{
|
236 |
+
"epoch": 0.66,
|
237 |
+
"grad_norm": 3.570347785949707,
|
238 |
+
"learning_rate": 2e-05,
|
239 |
+
"loss": 0.5628,
|
240 |
+
"step": 66
|
241 |
+
},
|
242 |
+
{
|
243 |
+
"epoch": 0.68,
|
244 |
+
"grad_norm": 6.035056114196777,
|
245 |
+
"learning_rate": 2e-05,
|
246 |
+
"loss": 0.8242,
|
247 |
+
"step": 68
|
248 |
+
},
|
249 |
+
{
|
250 |
+
"epoch": 0.7,
|
251 |
+
"grad_norm": 2.1205101013183594,
|
252 |
+
"learning_rate": 2e-05,
|
253 |
+
"loss": 0.9068,
|
254 |
+
"step": 70
|
255 |
+
},
|
256 |
+
{
|
257 |
+
"epoch": 0.72,
|
258 |
+
"grad_norm": 3.7536263465881348,
|
259 |
+
"learning_rate": 2e-05,
|
260 |
+
"loss": 1.7733,
|
261 |
+
"step": 72
|
262 |
+
},
|
263 |
+
{
|
264 |
+
"epoch": 0.74,
|
265 |
+
"grad_norm": 5.084173679351807,
|
266 |
+
"learning_rate": 2e-05,
|
267 |
+
"loss": 1.139,
|
268 |
+
"step": 74
|
269 |
+
},
|
270 |
+
{
|
271 |
+
"epoch": 0.76,
|
272 |
+
"grad_norm": 0.9449111819267273,
|
273 |
+
"learning_rate": 2e-05,
|
274 |
+
"loss": 0.5757,
|
275 |
+
"step": 76
|
276 |
+
},
|
277 |
+
{
|
278 |
+
"epoch": 0.78,
|
279 |
+
"grad_norm": 3.6127724647521973,
|
280 |
+
"learning_rate": 2e-05,
|
281 |
+
"loss": 1.3481,
|
282 |
+
"step": 78
|
283 |
+
},
|
284 |
+
{
|
285 |
+
"epoch": 0.8,
|
286 |
+
"grad_norm": 1.3057523965835571,
|
287 |
+
"learning_rate": 2e-05,
|
288 |
+
"loss": 0.705,
|
289 |
+
"step": 80
|
290 |
+
},
|
291 |
+
{
|
292 |
+
"epoch": 0.82,
|
293 |
+
"grad_norm": 1.5335392951965332,
|
294 |
+
"learning_rate": 2e-05,
|
295 |
+
"loss": 0.9568,
|
296 |
+
"step": 82
|
297 |
+
},
|
298 |
+
{
|
299 |
+
"epoch": 0.84,
|
300 |
+
"grad_norm": 2.186148166656494,
|
301 |
+
"learning_rate": 2e-05,
|
302 |
+
"loss": 0.5098,
|
303 |
+
"step": 84
|
304 |
+
},
|
305 |
+
{
|
306 |
+
"epoch": 0.86,
|
307 |
+
"grad_norm": 4.127236843109131,
|
308 |
+
"learning_rate": 2e-05,
|
309 |
+
"loss": 2.0965,
|
310 |
+
"step": 86
|
311 |
+
},
|
312 |
+
{
|
313 |
+
"epoch": 0.88,
|
314 |
+
"grad_norm": 2.2698607444763184,
|
315 |
+
"learning_rate": 2e-05,
|
316 |
+
"loss": 1.8116,
|
317 |
+
"step": 88
|
318 |
+
},
|
319 |
+
{
|
320 |
+
"epoch": 0.9,
|
321 |
+
"grad_norm": 7.15731143951416,
|
322 |
+
"learning_rate": 2e-05,
|
323 |
+
"loss": 2.0281,
|
324 |
+
"step": 90
|
325 |
+
},
|
326 |
+
{
|
327 |
+
"epoch": 0.92,
|
328 |
+
"grad_norm": 1.700381875038147,
|
329 |
+
"learning_rate": 2e-05,
|
330 |
+
"loss": 1.3664,
|
331 |
+
"step": 92
|
332 |
+
},
|
333 |
+
{
|
334 |
+
"epoch": 0.94,
|
335 |
+
"grad_norm": 0.8054038286209106,
|
336 |
+
"learning_rate": 2e-05,
|
337 |
+
"loss": 1.0307,
|
338 |
+
"step": 94
|
339 |
+
},
|
340 |
+
{
|
341 |
+
"epoch": 0.96,
|
342 |
+
"grad_norm": 2.1921820640563965,
|
343 |
+
"learning_rate": 2e-05,
|
344 |
+
"loss": 1.1617,
|
345 |
+
"step": 96
|
346 |
+
},
|
347 |
+
{
|
348 |
+
"epoch": 0.98,
|
349 |
+
"grad_norm": 2.8828494548797607,
|
350 |
+
"learning_rate": 2e-05,
|
351 |
+
"loss": 0.7795,
|
352 |
+
"step": 98
|
353 |
+
},
|
354 |
+
{
|
355 |
+
"epoch": 1.0,
|
356 |
+
"grad_norm": 1.916834831237793,
|
357 |
+
"learning_rate": 2e-05,
|
358 |
+
"loss": 0.7422,
|
359 |
+
"step": 100
|
360 |
+
},
|
361 |
+
{
|
362 |
+
"epoch": 1.0,
|
363 |
+
"step": 100,
|
364 |
+
"total_flos": 2233291239325696.0,
|
365 |
+
"train_loss": 1.2525035691261293,
|
366 |
+
"train_runtime": 160.8169,
|
367 |
+
"train_samples_per_second": 2.487,
|
368 |
+
"train_steps_per_second": 0.622
|
369 |
+
}
|
370 |
+
],
|
371 |
+
"logging_steps": 2,
|
372 |
+
"max_steps": 100,
|
373 |
+
"num_input_tokens_seen": 0,
|
374 |
+
"num_train_epochs": 1,
|
375 |
+
"save_steps": 500,
|
376 |
+
"stateful_callbacks": {
|
377 |
+
"TrainerControl": {
|
378 |
+
"args": {
|
379 |
+
"should_epoch_stop": false,
|
380 |
+
"should_evaluate": false,
|
381 |
+
"should_log": false,
|
382 |
+
"should_save": false,
|
383 |
+
"should_training_stop": false
|
384 |
+
},
|
385 |
+
"attributes": {}
|
386 |
+
}
|
387 |
+
},
|
388 |
+
"total_flos": 2233291239325696.0,
|
389 |
+
"train_batch_size": 1,
|
390 |
+
"trial_name": null,
|
391 |
+
"trial_params": null
|
392 |
+
}
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/4_client_model_round10.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c88bcdbcef96f0cc9f10b056f032824471bf31525402a1888c54d6a5bd93ddf6
|
3 |
+
size 778341886
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/4_client_model_round12.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:78f69654677d277697dcf550b91d3448ea03a05c37dfd28e0723874e20adc908
|
3 |
+
size 778341886
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/4_client_model_round15.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:95b29534740cdf69dfee373ef8bc9c0749923bde13b5208006fbc18916f3ca0a
|
3 |
+
size 778341886
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/4_client_model_round17.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:86cfeae9f0b9a252446507106e41bbcc13e74bcb4cc11f9f5e173506fd0dda79
|
3 |
+
size 778341886
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/4_client_model_round2.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1ef777dc711277e61509e50d38a953a057667fa139be8faeac789ba5c2698a3f
|
3 |
+
size 778341034
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/4_client_model_round20.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:eb30030ae9a34e79c71b04a8f44f1913824dc261d37d6b9c6c53b09aa4fe5d70
|
3 |
+
size 778341886
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/4_client_model_round5.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3ec26c19913afffba6c0fd3585819787a6c533aa5c831d9a9daef84aba1bcc71
|
3 |
+
size 778341034
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/4_client_model_round7.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:723a4d97f24e5c1a311901f6621474755cd35e8326fff914074f1a7e1f399beb
|
3 |
+
size 778341034
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/4_trainer_state.json
ADDED
@@ -0,0 +1,392 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"best_metric": null,
|
3 |
+
"best_model_checkpoint": null,
|
4 |
+
"epoch": 1.0,
|
5 |
+
"eval_steps": 500,
|
6 |
+
"global_step": 100,
|
7 |
+
"is_hyper_param_search": false,
|
8 |
+
"is_local_process_zero": true,
|
9 |
+
"is_world_process_zero": true,
|
10 |
+
"log_history": [
|
11 |
+
{
|
12 |
+
"epoch": 0.02,
|
13 |
+
"grad_norm": 1.3815970420837402,
|
14 |
+
"learning_rate": 2e-05,
|
15 |
+
"loss": 0.5863,
|
16 |
+
"step": 2
|
17 |
+
},
|
18 |
+
{
|
19 |
+
"epoch": 0.04,
|
20 |
+
"grad_norm": 2.8185386657714844,
|
21 |
+
"learning_rate": 2e-05,
|
22 |
+
"loss": 1.2971,
|
23 |
+
"step": 4
|
24 |
+
},
|
25 |
+
{
|
26 |
+
"epoch": 0.06,
|
27 |
+
"grad_norm": 2.8948168754577637,
|
28 |
+
"learning_rate": 2e-05,
|
29 |
+
"loss": 0.9846,
|
30 |
+
"step": 6
|
31 |
+
},
|
32 |
+
{
|
33 |
+
"epoch": 0.08,
|
34 |
+
"grad_norm": 4.101174831390381,
|
35 |
+
"learning_rate": 2e-05,
|
36 |
+
"loss": 1.688,
|
37 |
+
"step": 8
|
38 |
+
},
|
39 |
+
{
|
40 |
+
"epoch": 0.1,
|
41 |
+
"grad_norm": 2.688793897628784,
|
42 |
+
"learning_rate": 2e-05,
|
43 |
+
"loss": 0.6859,
|
44 |
+
"step": 10
|
45 |
+
},
|
46 |
+
{
|
47 |
+
"epoch": 0.12,
|
48 |
+
"grad_norm": 3.47356915473938,
|
49 |
+
"learning_rate": 2e-05,
|
50 |
+
"loss": 1.0135,
|
51 |
+
"step": 12
|
52 |
+
},
|
53 |
+
{
|
54 |
+
"epoch": 0.14,
|
55 |
+
"grad_norm": 3.2694878578186035,
|
56 |
+
"learning_rate": 2e-05,
|
57 |
+
"loss": 0.9047,
|
58 |
+
"step": 14
|
59 |
+
},
|
60 |
+
{
|
61 |
+
"epoch": 0.16,
|
62 |
+
"grad_norm": 1.625256061553955,
|
63 |
+
"learning_rate": 2e-05,
|
64 |
+
"loss": 1.125,
|
65 |
+
"step": 16
|
66 |
+
},
|
67 |
+
{
|
68 |
+
"epoch": 0.18,
|
69 |
+
"grad_norm": 2.0847363471984863,
|
70 |
+
"learning_rate": 2e-05,
|
71 |
+
"loss": 1.2551,
|
72 |
+
"step": 18
|
73 |
+
},
|
74 |
+
{
|
75 |
+
"epoch": 0.2,
|
76 |
+
"grad_norm": 1.9543544054031372,
|
77 |
+
"learning_rate": 2e-05,
|
78 |
+
"loss": 0.7905,
|
79 |
+
"step": 20
|
80 |
+
},
|
81 |
+
{
|
82 |
+
"epoch": 0.22,
|
83 |
+
"grad_norm": 2.1540379524230957,
|
84 |
+
"learning_rate": 2e-05,
|
85 |
+
"loss": 0.7426,
|
86 |
+
"step": 22
|
87 |
+
},
|
88 |
+
{
|
89 |
+
"epoch": 0.24,
|
90 |
+
"grad_norm": 0.8286647200584412,
|
91 |
+
"learning_rate": 2e-05,
|
92 |
+
"loss": 0.3257,
|
93 |
+
"step": 24
|
94 |
+
},
|
95 |
+
{
|
96 |
+
"epoch": 0.26,
|
97 |
+
"grad_norm": 1.9040488004684448,
|
98 |
+
"learning_rate": 2e-05,
|
99 |
+
"loss": 0.5483,
|
100 |
+
"step": 26
|
101 |
+
},
|
102 |
+
{
|
103 |
+
"epoch": 0.28,
|
104 |
+
"grad_norm": 1.71846342086792,
|
105 |
+
"learning_rate": 2e-05,
|
106 |
+
"loss": 0.7243,
|
107 |
+
"step": 28
|
108 |
+
},
|
109 |
+
{
|
110 |
+
"epoch": 0.3,
|
111 |
+
"grad_norm": 2.20640230178833,
|
112 |
+
"learning_rate": 2e-05,
|
113 |
+
"loss": 1.1453,
|
114 |
+
"step": 30
|
115 |
+
},
|
116 |
+
{
|
117 |
+
"epoch": 0.32,
|
118 |
+
"grad_norm": 3.0493052005767822,
|
119 |
+
"learning_rate": 2e-05,
|
120 |
+
"loss": 1.1843,
|
121 |
+
"step": 32
|
122 |
+
},
|
123 |
+
{
|
124 |
+
"epoch": 0.34,
|
125 |
+
"grad_norm": 1.9781206846237183,
|
126 |
+
"learning_rate": 2e-05,
|
127 |
+
"loss": 0.8867,
|
128 |
+
"step": 34
|
129 |
+
},
|
130 |
+
{
|
131 |
+
"epoch": 0.36,
|
132 |
+
"grad_norm": 2.0017993450164795,
|
133 |
+
"learning_rate": 2e-05,
|
134 |
+
"loss": 1.3522,
|
135 |
+
"step": 36
|
136 |
+
},
|
137 |
+
{
|
138 |
+
"epoch": 0.38,
|
139 |
+
"grad_norm": 1.269047737121582,
|
140 |
+
"learning_rate": 2e-05,
|
141 |
+
"loss": 0.6205,
|
142 |
+
"step": 38
|
143 |
+
},
|
144 |
+
{
|
145 |
+
"epoch": 0.4,
|
146 |
+
"grad_norm": 1.3758752346038818,
|
147 |
+
"learning_rate": 2e-05,
|
148 |
+
"loss": 0.6778,
|
149 |
+
"step": 40
|
150 |
+
},
|
151 |
+
{
|
152 |
+
"epoch": 0.42,
|
153 |
+
"grad_norm": 1.46341073513031,
|
154 |
+
"learning_rate": 2e-05,
|
155 |
+
"loss": 0.8367,
|
156 |
+
"step": 42
|
157 |
+
},
|
158 |
+
{
|
159 |
+
"epoch": 0.44,
|
160 |
+
"grad_norm": 3.903554916381836,
|
161 |
+
"learning_rate": 2e-05,
|
162 |
+
"loss": 1.1299,
|
163 |
+
"step": 44
|
164 |
+
},
|
165 |
+
{
|
166 |
+
"epoch": 0.46,
|
167 |
+
"grad_norm": 1.7048883438110352,
|
168 |
+
"learning_rate": 2e-05,
|
169 |
+
"loss": 0.388,
|
170 |
+
"step": 46
|
171 |
+
},
|
172 |
+
{
|
173 |
+
"epoch": 0.48,
|
174 |
+
"grad_norm": 1.586846947669983,
|
175 |
+
"learning_rate": 2e-05,
|
176 |
+
"loss": 1.0262,
|
177 |
+
"step": 48
|
178 |
+
},
|
179 |
+
{
|
180 |
+
"epoch": 0.5,
|
181 |
+
"grad_norm": 1.801108956336975,
|
182 |
+
"learning_rate": 2e-05,
|
183 |
+
"loss": 0.4058,
|
184 |
+
"step": 50
|
185 |
+
},
|
186 |
+
{
|
187 |
+
"epoch": 0.52,
|
188 |
+
"grad_norm": 3.518892765045166,
|
189 |
+
"learning_rate": 2e-05,
|
190 |
+
"loss": 1.8573,
|
191 |
+
"step": 52
|
192 |
+
},
|
193 |
+
{
|
194 |
+
"epoch": 0.54,
|
195 |
+
"grad_norm": 1.9927799701690674,
|
196 |
+
"learning_rate": 2e-05,
|
197 |
+
"loss": 0.8552,
|
198 |
+
"step": 54
|
199 |
+
},
|
200 |
+
{
|
201 |
+
"epoch": 0.56,
|
202 |
+
"grad_norm": 0.25938358902931213,
|
203 |
+
"learning_rate": 2e-05,
|
204 |
+
"loss": 0.5029,
|
205 |
+
"step": 56
|
206 |
+
},
|
207 |
+
{
|
208 |
+
"epoch": 0.58,
|
209 |
+
"grad_norm": 1.1329509019851685,
|
210 |
+
"learning_rate": 2e-05,
|
211 |
+
"loss": 0.6975,
|
212 |
+
"step": 58
|
213 |
+
},
|
214 |
+
{
|
215 |
+
"epoch": 0.6,
|
216 |
+
"grad_norm": 0.3482877016067505,
|
217 |
+
"learning_rate": 2e-05,
|
218 |
+
"loss": 0.269,
|
219 |
+
"step": 60
|
220 |
+
},
|
221 |
+
{
|
222 |
+
"epoch": 0.62,
|
223 |
+
"grad_norm": 0.7116028070449829,
|
224 |
+
"learning_rate": 2e-05,
|
225 |
+
"loss": 0.6903,
|
226 |
+
"step": 62
|
227 |
+
},
|
228 |
+
{
|
229 |
+
"epoch": 0.64,
|
230 |
+
"grad_norm": 2.584398031234741,
|
231 |
+
"learning_rate": 2e-05,
|
232 |
+
"loss": 1.0017,
|
233 |
+
"step": 64
|
234 |
+
},
|
235 |
+
{
|
236 |
+
"epoch": 0.66,
|
237 |
+
"grad_norm": 1.240759015083313,
|
238 |
+
"learning_rate": 2e-05,
|
239 |
+
"loss": 0.2974,
|
240 |
+
"step": 66
|
241 |
+
},
|
242 |
+
{
|
243 |
+
"epoch": 0.68,
|
244 |
+
"grad_norm": 1.373422622680664,
|
245 |
+
"learning_rate": 2e-05,
|
246 |
+
"loss": 0.9894,
|
247 |
+
"step": 68
|
248 |
+
},
|
249 |
+
{
|
250 |
+
"epoch": 0.7,
|
251 |
+
"grad_norm": 1.4504565000534058,
|
252 |
+
"learning_rate": 2e-05,
|
253 |
+
"loss": 1.0746,
|
254 |
+
"step": 70
|
255 |
+
},
|
256 |
+
{
|
257 |
+
"epoch": 0.72,
|
258 |
+
"grad_norm": 1.392308235168457,
|
259 |
+
"learning_rate": 2e-05,
|
260 |
+
"loss": 0.6376,
|
261 |
+
"step": 72
|
262 |
+
},
|
263 |
+
{
|
264 |
+
"epoch": 0.74,
|
265 |
+
"grad_norm": 1.5228718519210815,
|
266 |
+
"learning_rate": 2e-05,
|
267 |
+
"loss": 1.0094,
|
268 |
+
"step": 74
|
269 |
+
},
|
270 |
+
{
|
271 |
+
"epoch": 0.76,
|
272 |
+
"grad_norm": 0.7216625213623047,
|
273 |
+
"learning_rate": 2e-05,
|
274 |
+
"loss": 0.2343,
|
275 |
+
"step": 76
|
276 |
+
},
|
277 |
+
{
|
278 |
+
"epoch": 0.78,
|
279 |
+
"grad_norm": 0.6243200302124023,
|
280 |
+
"learning_rate": 2e-05,
|
281 |
+
"loss": 0.308,
|
282 |
+
"step": 78
|
283 |
+
},
|
284 |
+
{
|
285 |
+
"epoch": 0.8,
|
286 |
+
"grad_norm": 6.7099385261535645,
|
287 |
+
"learning_rate": 2e-05,
|
288 |
+
"loss": 1.2533,
|
289 |
+
"step": 80
|
290 |
+
},
|
291 |
+
{
|
292 |
+
"epoch": 0.82,
|
293 |
+
"grad_norm": 1.371478796005249,
|
294 |
+
"learning_rate": 2e-05,
|
295 |
+
"loss": 1.7963,
|
296 |
+
"step": 82
|
297 |
+
},
|
298 |
+
{
|
299 |
+
"epoch": 0.84,
|
300 |
+
"grad_norm": 2.473755121231079,
|
301 |
+
"learning_rate": 2e-05,
|
302 |
+
"loss": 0.7662,
|
303 |
+
"step": 84
|
304 |
+
},
|
305 |
+
{
|
306 |
+
"epoch": 0.86,
|
307 |
+
"grad_norm": 2.194167137145996,
|
308 |
+
"learning_rate": 2e-05,
|
309 |
+
"loss": 1.0925,
|
310 |
+
"step": 86
|
311 |
+
},
|
312 |
+
{
|
313 |
+
"epoch": 0.88,
|
314 |
+
"grad_norm": 2.4907443523406982,
|
315 |
+
"learning_rate": 2e-05,
|
316 |
+
"loss": 1.1646,
|
317 |
+
"step": 88
|
318 |
+
},
|
319 |
+
{
|
320 |
+
"epoch": 0.9,
|
321 |
+
"grad_norm": 1.6863021850585938,
|
322 |
+
"learning_rate": 2e-05,
|
323 |
+
"loss": 0.6589,
|
324 |
+
"step": 90
|
325 |
+
},
|
326 |
+
{
|
327 |
+
"epoch": 0.92,
|
328 |
+
"grad_norm": 1.869765043258667,
|
329 |
+
"learning_rate": 2e-05,
|
330 |
+
"loss": 0.9901,
|
331 |
+
"step": 92
|
332 |
+
},
|
333 |
+
{
|
334 |
+
"epoch": 0.94,
|
335 |
+
"grad_norm": 0.8603519201278687,
|
336 |
+
"learning_rate": 2e-05,
|
337 |
+
"loss": 0.4007,
|
338 |
+
"step": 94
|
339 |
+
},
|
340 |
+
{
|
341 |
+
"epoch": 0.96,
|
342 |
+
"grad_norm": 0.9052290320396423,
|
343 |
+
"learning_rate": 2e-05,
|
344 |
+
"loss": 0.6978,
|
345 |
+
"step": 96
|
346 |
+
},
|
347 |
+
{
|
348 |
+
"epoch": 0.98,
|
349 |
+
"grad_norm": 1.2907326221466064,
|
350 |
+
"learning_rate": 2e-05,
|
351 |
+
"loss": 0.4787,
|
352 |
+
"step": 98
|
353 |
+
},
|
354 |
+
{
|
355 |
+
"epoch": 1.0,
|
356 |
+
"grad_norm": 1.4034663438796997,
|
357 |
+
"learning_rate": 2e-05,
|
358 |
+
"loss": 0.6854,
|
359 |
+
"step": 100
|
360 |
+
},
|
361 |
+
{
|
362 |
+
"epoch": 1.0,
|
363 |
+
"step": 100,
|
364 |
+
"total_flos": 6020085697216512.0,
|
365 |
+
"train_loss": 0.8546864128112793,
|
366 |
+
"train_runtime": 255.754,
|
367 |
+
"train_samples_per_second": 1.564,
|
368 |
+
"train_steps_per_second": 0.391
|
369 |
+
}
|
370 |
+
],
|
371 |
+
"logging_steps": 2,
|
372 |
+
"max_steps": 100,
|
373 |
+
"num_input_tokens_seen": 0,
|
374 |
+
"num_train_epochs": 1,
|
375 |
+
"save_steps": 500,
|
376 |
+
"stateful_callbacks": {
|
377 |
+
"TrainerControl": {
|
378 |
+
"args": {
|
379 |
+
"should_epoch_stop": false,
|
380 |
+
"should_evaluate": false,
|
381 |
+
"should_log": false,
|
382 |
+
"should_save": false,
|
383 |
+
"should_training_stop": false
|
384 |
+
},
|
385 |
+
"attributes": {}
|
386 |
+
}
|
387 |
+
},
|
388 |
+
"total_flos": 6020085697216512.0,
|
389 |
+
"train_batch_size": 1,
|
390 |
+
"trial_name": null,
|
391 |
+
"trial_params": null
|
392 |
+
}
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/5_client_model_round10.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:16c3a1b765d8bd5932df4956901960360e5b4e813af8c4fd48a911511d54f299
|
3 |
+
size 778341886
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/5_client_model_round12.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c4a002bd6bd961d702a9f40536b342cae52b650251facf614e81064f85fe5901
|
3 |
+
size 778341886
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/5_client_model_round15.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:774bae98e390e5a37e10de5e0db1f24e9e86a534077998f010879dc7d6b6e3d3
|
3 |
+
size 778341886
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/5_client_model_round17.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a5e7450aea0bcdceccc04163ce77df3ae35350518486063118edac379c7c8470
|
3 |
+
size 778341886
|
client_states_feddpa_bs4_saveoptim_lr2e-5_sc315_4tasks_5rounds_fixitr100_T0125_decay099_SEED2/5_client_model_round2.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d61d2ee984ad9bac2c3a63e62f83964d12a7d10eec8862dfafbca85f8505c03b
|
3 |
+
size 778341034
|