serialexperimentsleon commited on
Commit
22d37cd
·
verified ·
1 Parent(s): e0e43c5

Add files using upload-large-folder tool

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. .gitattributes +132 -0
  2. config.yaml +598 -0
  3. videos/63438_0/epoch_20/0_traj_1000000_envseed.mp4 +3 -0
  4. videos/63438_0/epoch_20/10_traj_1000010_envseed.mp4 +3 -0
  5. videos/63438_0/epoch_20/11_traj_1000011_envseed.mp4 +3 -0
  6. videos/63438_0/epoch_20/12_traj_1000012_envseed.mp4 +3 -0
  7. videos/63438_0/epoch_20/13_traj_1000013_envseed.mp4 +3 -0
  8. videos/63438_0/epoch_20/14_traj_1000014_envseed.mp4 +3 -0
  9. videos/63438_0/epoch_20/15_traj_1000015_envseed.mp4 +3 -0
  10. videos/63438_0/epoch_20/16_traj_1000016_envseed.mp4 +3 -0
  11. videos/63438_0/epoch_20/17_traj_1000017_envseed.mp4 +3 -0
  12. videos/63438_0/epoch_20/18_traj_1000018_envseed.mp4 +3 -0
  13. videos/63438_0/epoch_20/19_traj_1000019_envseed.mp4 +3 -0
  14. videos/63438_0/epoch_20/1_traj_1000001_envseed.mp4 +3 -0
  15. videos/63438_0/epoch_20/20_traj_1000020_envseed.mp4 +3 -0
  16. videos/63438_0/epoch_20/21_traj_1000021_envseed.mp4 +3 -0
  17. videos/63438_0/epoch_20/22_traj_1000022_envseed.mp4 +3 -0
  18. videos/63438_0/epoch_20/23_traj_1000023_envseed.mp4 +3 -0
  19. videos/63438_0/epoch_20/24_traj_1000024_envseed.mp4 +3 -0
  20. videos/63438_0/epoch_20/25_traj_1000025_envseed.mp4 +3 -0
  21. videos/63438_0/epoch_20/26_traj_1000026_envseed.mp4 +3 -0
  22. videos/63438_0/epoch_20/27_traj_1000027_envseed.mp4 +3 -0
  23. videos/63438_0/epoch_20/28_traj_1000028_envseed.mp4 +3 -0
  24. videos/63438_0/epoch_20/29_traj_1000029_envseed.mp4 +3 -0
  25. videos/63438_0/epoch_20/2_traj_1000002_envseed.mp4 +3 -0
  26. videos/63438_0/epoch_20/30_traj_1000030_envseed.mp4 +3 -0
  27. videos/63438_0/epoch_20/31_traj_1000031_envseed.mp4 +3 -0
  28. videos/63438_0/epoch_20/32_traj_1000032_envseed.mp4 +3 -0
  29. videos/63438_0/epoch_20/33_traj_1000033_envseed.mp4 +3 -0
  30. videos/63438_0/epoch_20/34_traj_1000034_envseed.mp4 +3 -0
  31. videos/63438_0/epoch_20/35_traj_1000035_envseed.mp4 +3 -0
  32. videos/63438_0/epoch_20/36_traj_1000036_envseed.mp4 +3 -0
  33. videos/63438_0/epoch_20/37_traj_1000037_envseed.mp4 +3 -0
  34. videos/63438_0/epoch_20/38_traj_1000038_envseed.mp4 +3 -0
  35. videos/63438_0/epoch_20/39_traj_1000039_envseed.mp4 +3 -0
  36. videos/63438_0/epoch_20/3_traj_1000003_envseed.mp4 +3 -0
  37. videos/63438_0/epoch_20/40_traj_1000040_envseed.mp4 +3 -0
  38. videos/63438_0/epoch_20/41_traj_1000041_envseed.mp4 +3 -0
  39. videos/63438_0/epoch_20/42_traj_1000042_envseed.mp4 +3 -0
  40. videos/63438_0/epoch_20/43_traj_1000043_envseed.mp4 +3 -0
  41. videos/63438_0/epoch_20/44_traj_1000044_envseed.mp4 +3 -0
  42. videos/63438_0/epoch_20/45_traj_1000045_envseed.mp4 +3 -0
  43. videos/63438_0/epoch_20/46_traj_1000046_envseed.mp4 +3 -0
  44. videos/63438_0/epoch_20/47_traj_1000047_envseed.mp4 +3 -0
  45. videos/63438_0/epoch_20/48_traj_1000048_envseed.mp4 +3 -0
  46. videos/63438_0/epoch_20/49_traj_1000049_envseed.mp4 +3 -0
  47. videos/63438_0/epoch_20/4_traj_1000004_envseed.mp4 +3 -0
  48. videos/63438_0/epoch_20/50_traj_1000050_envseed.mp4 +3 -0
  49. videos/63438_0/epoch_20/51_traj_1000051_envseed.mp4 +3 -0
  50. videos/63438_0/epoch_20/52_traj_1000052_envseed.mp4 +3 -0
.gitattributes CHANGED
@@ -33,3 +33,135 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ videos/63438_0/epoch_20/97_traj_1000097_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
37
+ videos/63438_0/epoch_20/28_traj_1000028_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
38
+ videos/63438_0/epoch_20/15_traj_1000015_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
39
+ videos/63438_0/epoch_20/14_traj_1000014_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
40
+ videos/63438_0/epoch_20/95_traj_1000095_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
41
+ videos/63438_0/epoch_20/35_traj_1000035_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
42
+ videos/63438_0/epoch_20/81_traj_1000081_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
43
+ videos/63438_0/epoch_20/65_traj_1000065_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
44
+ videos/63438_0/epoch_20/75_traj_1000075_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
45
+ videos/63438_0/epoch_20/37_traj_1000037_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
46
+ videos/63438_0/epoch_20/19_traj_1000019_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
47
+ videos/63438_0/epoch_20/46_traj_1000046_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
48
+ videos/63438_0/epoch_20/44_traj_1000044_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
49
+ videos/63438_0/epoch_20/89_traj_1000089_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
50
+ videos/63438_0/epoch_20/11_traj_1000011_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
51
+ videos/63438_0/epoch_20/67_traj_1000067_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
52
+ videos/63438_0/epoch_20/51_traj_1000051_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
53
+ videos/63438_0/epoch_20/21_traj_1000021_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
54
+ videos/63438_0/epoch_20/9_traj_1000009_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
55
+ videos/63438_0/epoch_20/90_traj_1000090_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
56
+ videos/63438_0/epoch_20/10_traj_1000010_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
57
+ videos/63438_0/epoch_20/79_traj_1000079_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
58
+ videos/63438_0/epoch_20/43_traj_1000043_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
59
+ videos/63438_0/epoch_20/55_traj_1000055_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
60
+ videos/63438_0/epoch_20/30_traj_1000030_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
61
+ videos/63438_0/epoch_20/33_traj_1000033_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
62
+ videos/63438_0/epoch_20/29_traj_1000029_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
63
+ videos/63438_0/epoch_20/74_traj_1000074_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
64
+ videos/63438_0/epoch_20/54_traj_1000054_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
65
+ videos/63438_0/epoch_20/57_traj_1000057_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
66
+ videos/63438_0/epoch_20/64_traj_1000064_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
67
+ videos/63438_0/epoch_20/82_traj_1000082_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
68
+ videos/63438_0/epoch_20/70_traj_1000070_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
69
+ videos/63438_0/epoch_20/80_traj_1000080_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
70
+ videos/63438_0/epoch_20/92_traj_1000092_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
71
+ videos/63438_0/epoch_20/98_traj_1000098_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
72
+ videos/63438_0/epoch_20/53_traj_1000053_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
73
+ videos/63438_0/epoch_20/62_traj_1000062_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
74
+ videos/63438_0/epoch_20/22_traj_1000022_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
75
+ videos/63438_0/epoch_20/93_traj_1000093_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
76
+ videos/63438_0/epoch_20/0_traj_1000000_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
77
+ videos/63438_0/epoch_20/73_traj_1000073_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
78
+ videos/63438_0/epoch_20/48_traj_1000048_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
79
+ videos/63438_0/epoch_20/8_traj_1000008_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
80
+ videos/63438_0/epoch_20/60_traj_1000060_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
81
+ videos/63438_0/epoch_20/99_traj_1000099_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
82
+ videos/63438_0/epoch_20/39_traj_1000039_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
83
+ videos/63438_0/epoch_20/20_traj_1000020_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
84
+ videos/63438_0/epoch_20/2_traj_1000002_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
85
+ videos/63438_0/epoch_20/1_traj_1000001_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
86
+ videos/63438_0/epoch_20/24_traj_1000024_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
87
+ videos/63438_0/epoch_20/49_traj_1000049_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
88
+ videos/63438_0/epoch_20/63_traj_1000063_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
89
+ videos/63438_0/epoch_20/72_traj_1000072_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
90
+ videos/63438_0/epoch_20/45_traj_1000045_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
91
+ videos/63438_0/epoch_20/68_traj_1000068_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
92
+ videos/63438_0/epoch_20/59_traj_1000059_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
93
+ videos/63438_0/epoch_20/17_traj_1000017_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
94
+ videos/63438_0/epoch_20/61_traj_1000061_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
95
+ videos/63438_0/epoch_20/94_traj_1000094_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
96
+ videos/63438_0/epoch_20/69_traj_1000069_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
97
+ videos/63438_0/epoch_20/36_traj_1000036_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
98
+ videos/63438_0/epoch_20/85_traj_1000085_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
99
+ videos/63438_0/epoch_20/66_traj_1000066_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
100
+ videos/63438_0/epoch_20/84_traj_1000084_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
101
+ videos/63438_0/epoch_20/31_traj_1000031_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
102
+ videos/63438_0/epoch_20/41_traj_1000041_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
103
+ videos/63438_0/epoch_20/91_traj_1000091_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
104
+ videos/63438_0/epoch_20/83_traj_1000083_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
105
+ videos/63438_0/epoch_20/7_traj_1000007_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
106
+ videos/63438_0/epoch_20/96_traj_1000096_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
107
+ videos/63438_0/epoch_20/18_traj_1000018_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
108
+ videos/63438_0/epoch_20/88_traj_1000088_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
109
+ videos/63438_0/epoch_20/5_traj_1000005_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
110
+ videos/63438_0/epoch_20/12_traj_1000012_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
111
+ videos/63438_0/epoch_20/56_traj_1000056_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
112
+ videos/63438_0/epoch_20/38_traj_1000038_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
113
+ videos/63438_0/epoch_20/40_traj_1000040_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
114
+ videos/63438_0/epoch_20/16_traj_1000016_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
115
+ videos/63438_0/epoch_20/52_traj_1000052_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
116
+ videos/63438_0/epoch_20/26_traj_1000026_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
117
+ videos/63438_0/epoch_20/78_traj_1000078_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
118
+ videos/63438_0/epoch_20/13_traj_1000013_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
119
+ videos/63438_0/epoch_20/3_traj_1000003_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
120
+ videos/63438_0/epoch_20/87_traj_1000087_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
121
+ videos/63438_0/epoch_20/76_traj_1000076_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
122
+ videos/63438_0/epoch_20/32_traj_1000032_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
123
+ videos/63438_0/epoch_20/4_traj_1000004_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
124
+ videos/63438_0/epoch_20/50_traj_1000050_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
125
+ videos/63438_0/epoch_20/34_traj_1000034_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
126
+ videos/63438_0/epoch_20/25_traj_1000025_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
127
+ videos/63438_0/epoch_20/23_traj_1000023_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
128
+ videos/63438_0/epoch_20/42_traj_1000042_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
129
+ videos/63438_0/epoch_20/27_traj_1000027_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
130
+ videos/63438_0/epoch_20/71_traj_1000071_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
131
+ videos/63438_0/epoch_80/351_traj_1000051_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
132
+ videos/63438_0/epoch_20/47_traj_1000047_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
133
+ videos/63438_0/epoch_20/58_traj_1000058_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
134
+ videos/63438_0/epoch_80/357_traj_1000057_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
135
+ videos/63438_0/epoch_80/345_traj_1000045_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
136
+ videos/63438_0/epoch_80/350_traj_1000050_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
137
+ videos/63438_0/epoch_80/330_traj_1000030_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
138
+ videos/63438_0/epoch_80/392_traj_1000092_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
139
+ videos/63438_0/epoch_80/372_traj_1000072_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
140
+ videos/63438_0/epoch_20/6_traj_1000006_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
141
+ videos/63438_0/epoch_80/347_traj_1000047_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
142
+ videos/63438_0/epoch_20/77_traj_1000077_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
143
+ videos/63438_0/epoch_80/399_traj_1000099_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
144
+ videos/63438_0/epoch_80/343_traj_1000043_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
145
+ videos/63438_0/epoch_80/336_traj_1000036_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
146
+ videos/63438_0/epoch_80/370_traj_1000070_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
147
+ videos/63438_0/epoch_80/356_traj_1000056_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
148
+ videos/63438_0/epoch_20/86_traj_1000086_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
149
+ videos/63438_0/epoch_80/308_traj_1000008_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
150
+ videos/63438_0/epoch_80/346_traj_1000046_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
151
+ videos/63438_0/epoch_80/307_traj_1000007_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
152
+ videos/63438_0/epoch_80/388_traj_1000088_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
153
+ videos/63438_0/epoch_80/332_traj_1000032_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
154
+ videos/63438_0/epoch_80/325_traj_1000025_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
155
+ videos/63438_0/epoch_80/337_traj_1000037_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
156
+ videos/63438_0/epoch_80/379_traj_1000079_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
157
+ videos/63438_0/epoch_80/327_traj_1000027_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
158
+ videos/63438_0/epoch_80/302_traj_1000002_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
159
+ videos/63438_0/epoch_80/333_traj_1000033_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
160
+ videos/63438_0/epoch_80/359_traj_1000059_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
161
+ videos/63438_0/epoch_80/375_traj_1000075_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
162
+ videos/63438_0/epoch_80/326_traj_1000026_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
163
+ videos/63438_0/epoch_80/319_traj_1000019_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
164
+ videos/63438_0/epoch_80/371_traj_1000071_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
165
+ videos/63438_0/epoch_80/301_traj_1000001_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
166
+ videos/63438_0/epoch_80/398_traj_1000098_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
167
+ videos/63438_0/epoch_80/305_traj_1000005_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
config.yaml ADDED
@@ -0,0 +1,598 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ root_dir: /mnt/kostas-graid/datasets/extrinsic_contact_data
2
+ replay_buffer_size: 150000
3
+ replay_buffer_num_workers: 2
4
+ nstep: 3
5
+ batch_size: 128
6
+ seed: 0
7
+ dataset_shuffle_seed: ${seed}
8
+ valid_dataset_shuffle_seed: 0
9
+ train_dataset_shuffle_seed: 0
10
+ device: cuda
11
+ save_video: true
12
+ save_train_video: true
13
+ use_tb: true
14
+ use_wandb: true
15
+ wandb_run_id: '63438_0'
16
+ wandb_notes: 63438_0_restarted_1
17
+ eval: false
18
+ eval_on_sim: true
19
+ eval_on_sim_past_checkpoints: false
20
+ simulation: true
21
+ true_action_history: false
22
+ train_pad_after: 4
23
+ process_contact_features: ${eval}
24
+ obs_type: pixels
25
+ use_color: true
26
+ use_depth: true
27
+ use_masks: false
28
+ mask_list:
29
+ - EE_obj_mask
30
+ mask_representation: channels
31
+ segmentation_model_name: gt_segmentation
32
+ crop_hw:
33
+ - 144
34
+ - 144
35
+ crop_down_offset: 48
36
+ color_crop_type: null
37
+ depth_crop_type: null
38
+ segmask_crop_type: null
39
+ add_crop_binary_mask: false
40
+ add_coord_conv_map: false
41
+ use_context_color: false
42
+ use_context_depth: false
43
+ use_context_segmask: false
44
+ context_color_crop_type: null
45
+ context_depth_crop_type: null
46
+ context_segmask_crop_type: null
47
+ context_add_crop_binary_mask: false
48
+ context_add_coord_conv_map: false
49
+ use_contact_map: false
50
+ use_sdf_maps: false
51
+ use_normals_maps: false
52
+ which_objects: both
53
+ max_contact_prob: 1.0
54
+ max_depth: 2.0
55
+ grasped_dtc_max_value: 0.2
56
+ env_dtc_max_value: 0.4
57
+ grasped_normals_mask_max_dtc_value: 0.2
58
+ env_normals_mask_max_dtc_value: 0.4
59
+ clamp_dtc: true
60
+ dtc_adaptive_normalization: false
61
+ mask_normals_within_sdf: true
62
+ adaptive_normals_mask: true
63
+ learnable_contact_preprocess_params: false
64
+ contact_model_name: gt_contact
65
+ contact_estimation_model_ckpt_path: ~/fish_leon/contact_estimation/artifacts/175604_2/checkpoints/epoch=09-val_loss=0.00.ckpt
66
+ encoder_type: small
67
+ debug_timestamps: false
68
+ open_loop: false
69
+ action_trajectories: true
70
+ stop_after_action: false
71
+ interpolation_frequency: 25
72
+ policy_frequency: 5
73
+ wait_for_new_camera_frames: true
74
+ baseline: false
75
+ use_num_train_epochs_to_determine_lr_schedule: true
76
+ spawn_new_env_books: false
77
+ spawn_new_grasped_book: true
78
+ shuffle_env_books_mode: none
79
+ book_ends_dict:
80
+ mode: dynamic
81
+ height: 0.1
82
+ mass: 3.5
83
+ friction: 0.15
84
+ color: '#808080'
85
+ record_sim_videos: true
86
+ resume_wandb_run: true
87
+ train_demo_idxs_list_or_num: -1
88
+ log_train_every_steps: 25
89
+ name_of_expert_demo: 240_sim_demos_left_of_4th_book_bookends_no_env_rand_20hz_act
90
+ expert_dataset_dirpath: ${root_dir}/FISH/expert_demos/${suite.name}/${task_name}/${name_of_expert_demo}
91
+ store_dataset_in_memory: false
92
+ expert_dataset: ${expert_dataset_dirpath}/demos.zarr
93
+ action_key: action
94
+ semantic_demo_grouping_name: semantic_demo_grouping.yaml
95
+ semantic_demo_grouping: ${expert_dataset_dirpath}/${semantic_demo_grouping_name}
96
+ include_groups_list: all
97
+ expert_dataset_config: ${expert_dataset_dirpath}/demo_config.yaml
98
+ name_of_valid_demo: 240_sim_demos_left_of_4th_book_bookends_no_env_rand_20hz_act
99
+ train_num_groups: 210
100
+ valid_dataset_dir: ${root_dir}/FISH/expert_demos/${suite.name}/${task_name}/${name_of_valid_demo}/demos.zarr
101
+ valid_demo_idxs_list_or_num: null
102
+ val_num_groups: 30
103
+ universal_unseen_env_seed_start: 1000000
104
+ load_bc: ${agent.load_checkpoint}
105
+ checkpoint_epoch_list:
106
+ - 99
107
+ - 199
108
+ - 299
109
+ - 399
110
+ - 499
111
+ - 599
112
+ - 699
113
+ - 799
114
+ - 899
115
+ - 999
116
+ - 1249
117
+ - 1499
118
+ - 1749
119
+ - 1999
120
+ - 2999
121
+ - 3999
122
+ - 4999
123
+ - 5999
124
+ - 6999
125
+ - 7999
126
+ - 8999
127
+ - 9999
128
+ snapshot_root_dir: /mnt/grasp_high_usage/leonmkim/contact_estimation/FISH
129
+ save_snapshot: true
130
+ save_last_snapshot: true
131
+ save_snapshot_when_done: true
132
+ top_k_checkpoints: 10
133
+ save_snapshot_link_to_weights_dir: deprecated
134
+ restore_missing_checkpoint: null
135
+ restore_and_eval_all_missing_checkpoints: false
136
+ bc_regularize: false
137
+ bc_weight_type: qfilter
138
+ username: leonmkim
139
+ hostname: grasp-login1
140
+ local_snapshot_root_dir: /mnt/bighdd/fish_contact_backup
141
+ local_snapshot_dir: ${local_snapshot_root_dir}/exp_local/${suite.name}_${obs_type}/${task_name}
142
+ experiment_dir: ./exp_local/${suite.name}_${obs_type}/${task_name}/${wandb_run_id}
143
+ agent:
144
+ _target_: agent.diffusion_policy.DiffusionPolicyAgent
145
+ name: diffusion_policy
146
+ load_checkpoint: ${eval}
147
+ device: ${device}
148
+ n_obs_steps: ${.config.policy_cfg.n_obs_steps}
149
+ suite_name: ${suite.name}
150
+ obs_type: ${obs_type}
151
+ enable_arm: ${eval}
152
+ enable_camera: ${eval}
153
+ use_tb: ${use_tb}
154
+ desired_image_shape:
155
+ - 13
156
+ - 180
157
+ - 240
158
+ orig_cam_shape:
159
+ - 3
160
+ - 240
161
+ - 320
162
+ actions_prenormalized: ${simulation}
163
+ config:
164
+ _target_: agent.diffusion_policy.DiffusionPolicyAgentConfig
165
+ compile: false
166
+ device: ${device}
167
+ cam_resize_shape: ${agent.desired_image_shape}
168
+ orig_cam_shape: ${agent.orig_cam_shape}
169
+ policy_frequency: ${policy_frequency}
170
+ interpolation_frequency: ${interpolation_frequency}
171
+ deterministic_actions: true
172
+ deterministic_actions_seed: 0
173
+ policy_cfg:
174
+ _target_: lerobot.common.policies.diffusion.configuration_diffusion.DiffusionConfig
175
+ n_obs_steps: 1
176
+ horizon: 28
177
+ n_action_steps: ${agent.config.policy_cfg.horizon}
178
+ output_shapes:
179
+ action:
180
+ - 7
181
+ input_normalization_modes:
182
+ observation.image: mean_std
183
+ observation.state: min_max
184
+ observation.action_history: min_max
185
+ output_normalization_modes:
186
+ action: min_max
187
+ vision_backbone: resnet18
188
+ crop_distractors_tlhw: null
189
+ pretrained_backbone_weights: null
190
+ transforms:
191
+ - _target_: torchaug.transforms.RandomAffine
192
+ degrees:
193
+ - -5
194
+ - 5
195
+ translate:
196
+ - 0.05
197
+ - 0.05
198
+ batch_transform: true
199
+ num_chunks: -1
200
+ batch_inplace: true
201
+ - _target_: torchvision.transforms.v2.Grayscale
202
+ num_output_channels: 3
203
+ use_group_norm: true
204
+ spatial_softmax_num_keypoints: 32
205
+ action_history_encoder_config:
206
+ _target_: lerobot.common.policies.diffusion.configuration_diffusion.Unet1dEncoderConfig
207
+ in_channels: 7
208
+ out_channels: 32
209
+ history_length: 4
210
+ kernel_size: ${agent.config.policy_cfg.kernel_size}
211
+ downsample_kernel_size: 3
212
+ downsample_stride: 2
213
+ downsample_padding: 1
214
+ down_dims:
215
+ - 128
216
+ - 256
217
+ - 512
218
+ kernel_size: 5
219
+ n_groups: 8
220
+ diffusion_step_embed_dim: 128
221
+ use_film_scale_modulation: true
222
+ noise_scheduler_type: DDIM
223
+ beta_schedule: squaredcos_cap_v2
224
+ beta_start: 0.0001
225
+ beta_end: 0.02
226
+ prediction_type: epsilon
227
+ clip_sample: true
228
+ clip_sample_range: 1.0
229
+ num_train_timesteps: 100
230
+ num_inference_steps: 100
231
+ do_mask_loss_for_padding: false
232
+ start_horizon_at_current_step: true
233
+ input_shapes:
234
+ observation.image:
235
+ - 13
236
+ - 180
237
+ - 240
238
+ context_observation.image:
239
+ - 13
240
+ - 180
241
+ - 240
242
+ observation.state:
243
+ - 8
244
+ observation.action_history:
245
+ - 7
246
+ train_cfg:
247
+ _target_: utils.TrainConfig
248
+ lr: 0.0001
249
+ lr_scheduler: cosine
250
+ lr_warmup_steps: 500
251
+ adam_betas:
252
+ - 0.95
253
+ - 0.999
254
+ adam_eps: 1.0e-08
255
+ adam_weight_decay: 1.0e-06
256
+ grad_clip_norm: 10
257
+ offline_steps: 303000
258
+ use_amp: true
259
+ optimizer: adamw
260
+ observation_cfg:
261
+ _target_: agent.encoder.VisualFeatureSet
262
+ use_depth: ${use_depth}
263
+ use_color: ${use_color}
264
+ mask_input_dict:
265
+ _target_: agent.encoder.MaskInputDict
266
+ enable: ${use_masks}
267
+ representation: ${mask_representation}
268
+ mask_list: ${mask_list}
269
+ segmentation_model_name: ${segmentation_model_name}
270
+ crop_input_config:
271
+ _target_: agent.encoder.CropInputConfig
272
+ color_crop_type: ${color_crop_type}
273
+ depth_crop_type: ${depth_crop_type}
274
+ segmask_crop_type: ${segmask_crop_type}
275
+ crop_hw: ${crop_hw}
276
+ crop_down_offset: ${crop_down_offset}
277
+ add_crop_binary_mask: ${add_crop_binary_mask}
278
+ add_coord_conv_map: ${add_coord_conv_map}
279
+ context_input_config:
280
+ _target_: agent.encoder.ContextInputConfig
281
+ use_color: ${use_context_color}
282
+ use_depth: ${use_context_depth}
283
+ mask_input_dict:
284
+ _target_: agent.encoder.MaskInputDict
285
+ enable: ${use_context_segmask}
286
+ representation: ${mask_representation}
287
+ mask_list: ${mask_list}
288
+ segmentation_model_name: ${segmentation_model_name}
289
+ crop_input_config:
290
+ _target_: agent.encoder.CropInputConfig
291
+ color_crop_type: ${context_color_crop_type}
292
+ depth_crop_type: ${context_depth_crop_type}
293
+ segmask_crop_type: ${context_segmask_crop_type}
294
+ crop_hw: ${crop_hw}
295
+ crop_down_offset: ${crop_down_offset}
296
+ add_crop_binary_mask: ${context_add_crop_binary_mask}
297
+ add_coord_conv_map: ${context_add_coord_conv_map}
298
+ mask_soft_approx_scheduler_config:
299
+ _target_: agent.encoder.MaskSoftApproxSchedulerConfig
300
+ num_steps: 40000
301
+ initial_value: 10.0
302
+ final_value: 1000.0
303
+ interpolation_scheme: cosine
304
+ use_contact_map: ${use_contact_map}
305
+ use_sdf_maps: ${use_sdf_maps}
306
+ use_normals_maps: ${use_normals_maps}
307
+ which_objects: ${which_objects}
308
+ grasped_dtc_max_value: ${grasped_dtc_max_value}
309
+ env_dtc_max_value: ${env_dtc_max_value}
310
+ grasped_normals_mask_max_dtc_value: ${grasped_normals_mask_max_dtc_value}
311
+ env_normals_mask_max_dtc_value: ${env_normals_mask_max_dtc_value}
312
+ clamp_dtc: ${clamp_dtc}
313
+ max_contact_prob: ${max_contact_prob}
314
+ mask_normals_within_sdf: ${mask_normals_within_sdf}
315
+ dtc_adaptive_normalization: ${dtc_adaptive_normalization}
316
+ adaptive_normals_mask: ${adaptive_normals_mask}
317
+ max_depth: ${max_depth}
318
+ image_shape: ${agent.desired_image_shape}
319
+ learnable_contact_preprocess_params: ${learnable_contact_preprocess_params}
320
+ learning_rate: 0.0001
321
+ weight_decay: 0.0
322
+ contact_model_name: ${contact_model_name}
323
+ zero_centered: false
324
+ crop_distractors_tlhw: ${agent.config.policy_cfg.crop_distractors_tlhw}
325
+ suite:
326
+ suite: frankagym
327
+ name: frankagym
328
+ frame_stack: ${agent.n_obs_steps}
329
+ action_repeat: 1
330
+ discount: 0.99
331
+ hidden_dim: 1024
332
+ num_train_frames: 1000000
333
+ num_seed_frames: 0
334
+ num_train_epochs: 1000
335
+ validate_every_epochs: 20
336
+ eval_on_sim_every_epochs: 20
337
+ validate_diffusion_on_action_loss_every_epochs: 20
338
+ eval_on_sim_every_epochs_diffusion: 500
339
+ train_eval_diffusion_on_action_loss_every_epochs: 20
340
+ check_topk_every_epochs: 20
341
+ save_snapshot_every_epochs: 25
342
+ num_seen_eval_envs: 0
343
+ num_unseen_eval_envs: 100
344
+ evals_per_env: 1
345
+ eval_every_frames: 2000
346
+ num_eval_episodes: 5
347
+ save_snapshot: true
348
+ wait_for_user_to_start_episode: true
349
+ task_make_fn:
350
+ _target_: suite.frankagym.make
351
+ name: ${task_name}
352
+ height: 240
353
+ width: 320
354
+ frame_stack: ${suite.frame_stack}
355
+ action_repeat: ${suite.action_repeat}
356
+ seed: ${seed}
357
+ enable_arm: ${agent.enable_arm}
358
+ enable_gripper: ${enable_gripper}
359
+ start_with_gripper_open: ${start_with_gripper_open}
360
+ enable_camera: ${agent.enable_camera}
361
+ path_to_depth_extrinsics: ${path_to_depth_extrinsics}
362
+ contact_estimation_model_ckpt_path: ${contact_estimation_model_ckpt_path}
363
+ x_limit: ${x_limit}
364
+ y_limit: ${y_limit}
365
+ z_limit: ${z_limit}
366
+ device: ${device}
367
+ interpolation_frequency: ${interpolation_frequency}
368
+ policy_frequency: ${policy_frequency}
369
+ debug_timestamps: ${debug_timestamps}
370
+ stop_after_action: ${stop_after_action}
371
+ open_loop: ${open_loop}
372
+ wait_for_new_camera_frames: ${wait_for_new_camera_frames}
373
+ action_key: ${action_key}
374
+ action_trajectory_horizon: ${agent.config.policy_cfg.horizon}
375
+ action_trajectories: ${action_trajectories}
376
+ path_to_zarr_dataset: ${expert_dataset}
377
+ agent_policy_cfg:
378
+ _target_: agent.diffusion_policy.DiffusionPolicyAgentConfig
379
+ compile: false
380
+ device: ${device}
381
+ cam_resize_shape: ${agent.desired_image_shape}
382
+ orig_cam_shape: ${agent.orig_cam_shape}
383
+ policy_frequency: ${policy_frequency}
384
+ interpolation_frequency: ${interpolation_frequency}
385
+ deterministic_actions: true
386
+ deterministic_actions_seed: 0
387
+ policy_cfg:
388
+ _target_: lerobot.common.policies.diffusion.configuration_diffusion.DiffusionConfig
389
+ n_obs_steps: 1
390
+ horizon: 28
391
+ n_action_steps: ${agent.config.policy_cfg.horizon}
392
+ output_shapes:
393
+ action:
394
+ - 7
395
+ input_normalization_modes:
396
+ observation.image: mean_std
397
+ observation.state: min_max
398
+ observation.action_history: min_max
399
+ output_normalization_modes:
400
+ action: min_max
401
+ vision_backbone: resnet18
402
+ crop_distractors_tlhw: null
403
+ pretrained_backbone_weights: null
404
+ transforms:
405
+ - _target_: torchaug.transforms.RandomAffine
406
+ degrees:
407
+ - -5
408
+ - 5
409
+ translate:
410
+ - 0.05
411
+ - 0.05
412
+ batch_transform: true
413
+ num_chunks: -1
414
+ batch_inplace: true
415
+ - _target_: torchvision.transforms.v2.Grayscale
416
+ num_output_channels: 3
417
+ use_group_norm: true
418
+ spatial_softmax_num_keypoints: 32
419
+ action_history_encoder_config:
420
+ _target_: lerobot.common.policies.diffusion.configuration_diffusion.Unet1dEncoderConfig
421
+ in_channels: 7
422
+ out_channels: 32
423
+ history_length: 4
424
+ kernel_size: ${agent.config.policy_cfg.kernel_size}
425
+ downsample_kernel_size: 3
426
+ downsample_stride: 2
427
+ downsample_padding: 1
428
+ down_dims:
429
+ - 128
430
+ - 256
431
+ - 512
432
+ kernel_size: 5
433
+ n_groups: 8
434
+ diffusion_step_embed_dim: 128
435
+ use_film_scale_modulation: true
436
+ noise_scheduler_type: DDIM
437
+ beta_schedule: squaredcos_cap_v2
438
+ beta_start: 0.0001
439
+ beta_end: 0.02
440
+ prediction_type: epsilon
441
+ clip_sample: true
442
+ clip_sample_range: 1.0
443
+ num_train_timesteps: 100
444
+ num_inference_steps: 100
445
+ do_mask_loss_for_padding: false
446
+ start_horizon_at_current_step: true
447
+ input_shapes:
448
+ observation.image:
449
+ - 13
450
+ - 180
451
+ - 240
452
+ context_observation.image:
453
+ - 13
454
+ - 180
455
+ - 240
456
+ observation.state:
457
+ - 8
458
+ observation.action_history:
459
+ - 7
460
+ train_cfg:
461
+ _target_: utils.TrainConfig
462
+ lr: 0.0001
463
+ lr_scheduler: cosine
464
+ lr_warmup_steps: 500
465
+ adam_betas:
466
+ - 0.95
467
+ - 0.999
468
+ adam_eps: 1.0e-08
469
+ adam_weight_decay: 1.0e-06
470
+ grad_clip_norm: 10
471
+ offline_steps: 303000
472
+ use_amp: true
473
+ optimizer: adamw
474
+ observation_cfg:
475
+ _target_: agent.encoder.VisualFeatureSet
476
+ use_depth: ${use_depth}
477
+ use_color: ${use_color}
478
+ mask_input_dict:
479
+ _target_: agent.encoder.MaskInputDict
480
+ enable: ${use_masks}
481
+ representation: ${mask_representation}
482
+ mask_list: ${mask_list}
483
+ segmentation_model_name: ${segmentation_model_name}
484
+ crop_input_config:
485
+ _target_: agent.encoder.CropInputConfig
486
+ color_crop_type: ${color_crop_type}
487
+ depth_crop_type: ${depth_crop_type}
488
+ segmask_crop_type: ${segmask_crop_type}
489
+ crop_hw: ${crop_hw}
490
+ crop_down_offset: ${crop_down_offset}
491
+ add_crop_binary_mask: ${add_crop_binary_mask}
492
+ add_coord_conv_map: ${add_coord_conv_map}
493
+ context_input_config:
494
+ _target_: agent.encoder.ContextInputConfig
495
+ use_color: ${use_context_color}
496
+ use_depth: ${use_context_depth}
497
+ mask_input_dict:
498
+ _target_: agent.encoder.MaskInputDict
499
+ enable: ${use_context_segmask}
500
+ representation: ${mask_representation}
501
+ mask_list: ${mask_list}
502
+ segmentation_model_name: ${segmentation_model_name}
503
+ crop_input_config:
504
+ _target_: agent.encoder.CropInputConfig
505
+ color_crop_type: ${context_color_crop_type}
506
+ depth_crop_type: ${context_depth_crop_type}
507
+ segmask_crop_type: ${context_segmask_crop_type}
508
+ crop_hw: ${crop_hw}
509
+ crop_down_offset: ${crop_down_offset}
510
+ add_crop_binary_mask: ${context_add_crop_binary_mask}
511
+ add_coord_conv_map: ${context_add_coord_conv_map}
512
+ mask_soft_approx_scheduler_config:
513
+ _target_: agent.encoder.MaskSoftApproxSchedulerConfig
514
+ num_steps: 40000
515
+ initial_value: 10.0
516
+ final_value: 1000.0
517
+ interpolation_scheme: cosine
518
+ use_contact_map: ${use_contact_map}
519
+ use_sdf_maps: ${use_sdf_maps}
520
+ use_normals_maps: ${use_normals_maps}
521
+ which_objects: ${which_objects}
522
+ grasped_dtc_max_value: ${grasped_dtc_max_value}
523
+ env_dtc_max_value: ${env_dtc_max_value}
524
+ grasped_normals_mask_max_dtc_value: ${grasped_normals_mask_max_dtc_value}
525
+ env_normals_mask_max_dtc_value: ${env_normals_mask_max_dtc_value}
526
+ clamp_dtc: ${clamp_dtc}
527
+ max_contact_prob: ${max_contact_prob}
528
+ mask_normals_within_sdf: ${mask_normals_within_sdf}
529
+ dtc_adaptive_normalization: ${dtc_adaptive_normalization}
530
+ adaptive_normals_mask: ${adaptive_normals_mask}
531
+ max_depth: ${max_depth}
532
+ image_shape: ${agent.desired_image_shape}
533
+ learnable_contact_preprocess_params: ${learnable_contact_preprocess_params}
534
+ learning_rate: 0.0001
535
+ weight_decay: 0.0
536
+ contact_model_name: ${contact_model_name}
537
+ zero_centered: false
538
+ crop_distractors_tlhw: ${agent.config.policy_cfg.crop_distractors_tlhw}
539
+ true_action_history: ${true_action_history}
540
+ num_train_frames_bc: 50000
541
+ num_train_frames_drq: 1100000
542
+ stddev_schedule_drq: linear(1.0,0.1,100000)
543
+ task_name: FrankaInsertion-v1
544
+ num_train_frames_vinn: 25000
545
+ num_train_frames_diffusion: 1000000
546
+ num_train_epochs_bc: 5000
547
+ num_train_epochs_diffusion: 1000
548
+ validate_every_epochs_bc: 5
549
+ validate_every_epochs_diffusion: 20
550
+ validate_diffusion_on_action_loss_every_epochs: 20
551
+ train_eval_diffusion_on_action_loss_every_epochs: 20
552
+ eval_on_sim_every_epochs_diffusion: 20
553
+ check_topk_every_epochs: 5
554
+ check_topk_every_epochs_diffusion: 20
555
+ save_snapshot_every_epochs_diffusion: 25
556
+ x_limit:
557
+ - 0.2
558
+ - 0.7
559
+ y_limit:
560
+ - -0.4
561
+ - 0.4
562
+ z_limit:
563
+ - -0.05
564
+ - 0.55
565
+ home_displacement:
566
+ - 0.55
567
+ - 0.0
568
+ - 0.55
569
+ - 180.0
570
+ - 0.0
571
+ - 0.0
572
+ enable_gripper: true
573
+ start_with_gripper_open: true
574
+ offset_mask:
575
+ - 1
576
+ - 1
577
+ - 1
578
+ - 1
579
+ - 1
580
+ - 1
581
+ path_to_depth_extrinsics: ~/fish_leon/FISH/cfgs/camera_poses/camera_poses_L515/20240904-122305/color_tf_world.npy
582
+ test:
583
+ username: leonmkim
584
+ hostname: grasp-login1
585
+ name_of_expert_demo: 112_240x320_all_twodim_left_to_right_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act
586
+ root_dir: ~/fish_leon
587
+ expert_dataset_dirpath: ${test.root_dir}/FISH/expert_demos/${suite.name}/${task_name}/${test.name_of_expert_demo}
588
+ expert_dataset: ${test.expert_dataset_dirpath}/demos.zarr
589
+ semantic_demo_grouping_name: semantic_demo_grouping.yaml
590
+ semantic_demo_grouping: ${test.expert_dataset_dirpath}/${test.semantic_demo_grouping_name}
591
+ include_groups_list: all
592
+ batch_size: 128
593
+ store_dataset_in_memory: false
594
+ use_tb: true
595
+ local_snapshot_root_dir: /mnt/bighdd/fish_contact_backup
596
+ local_snapshot_dir: ${test.local_snapshot_root_dir}/exp_local/${suite.name}_${obs_type}/${task_name}
597
+ resume_wandb_run: false
598
+ feature_type: 180x240_1_RGB_D_2.0_acthist_hst4_out32_dwnkrnl3_dwnstrd2_dwnpd1
videos/63438_0/epoch_20/0_traj_1000000_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ba62f0f8f6050d62edeb8a255633cc020c171efc1773da435da2bdf75da23042
3
+ size 356731
videos/63438_0/epoch_20/10_traj_1000010_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c791ebd37b8326398167f740ccc68107a2b4680dcaf1077a8666cb526d42925c
3
+ size 256816
videos/63438_0/epoch_20/11_traj_1000011_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:687d6941f62b4126025ee175179a6a1fab58ce296efd019ff59055d9d511ab84
3
+ size 389960
videos/63438_0/epoch_20/12_traj_1000012_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:95bdec2a175841d14258d566d9ba5b63515e338d8227f791e450e98374665393
3
+ size 334811
videos/63438_0/epoch_20/13_traj_1000013_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:31aaa319786dc42bdaae94c98fcbffdf31af1f4e21d3a3a94532d0bade4fb792
3
+ size 444864
videos/63438_0/epoch_20/14_traj_1000014_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9cfdde9ea97327ace1bbc10cdf9dc172188441cd48fa3d2007fc797534a60c00
3
+ size 280384
videos/63438_0/epoch_20/15_traj_1000015_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f099ed729c9705eb24f01bd3553334f6c709da81ca6253259de483e26cbfb928
3
+ size 245713
videos/63438_0/epoch_20/16_traj_1000016_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f17596bbb4b91077c7699f6bf173245db083797c35d0c05b47470869f27f2a58
3
+ size 337688
videos/63438_0/epoch_20/17_traj_1000017_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d2887c159e79c89860a87902e3022af077e06ed76c7d1e5ed135cbd11b24a4ba
3
+ size 231300
videos/63438_0/epoch_20/18_traj_1000018_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f13dc62c14252cd7183b4be8bc7220cbb894737ee87bd52972b6070bcd64bd86
3
+ size 421552
videos/63438_0/epoch_20/19_traj_1000019_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1086ddd8f87c2d5f812084a920f5a6370853eaf5a0f948f89c5a89b8fe7cf627
3
+ size 228387
videos/63438_0/epoch_20/1_traj_1000001_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:37390f319fa329bf37604a4845f108865932c2c191bed029cd1a7abf6bd9bedc
3
+ size 348605
videos/63438_0/epoch_20/20_traj_1000020_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6a6ef00fcf050c0545998b20aa6209461c6e73de0d8f05ff05527bf698928700
3
+ size 384558
videos/63438_0/epoch_20/21_traj_1000021_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d6210afa441240b8b76da08bf673267c86217bd6fb632778f8de09eca19d36ed
3
+ size 438968
videos/63438_0/epoch_20/22_traj_1000022_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3b20e447dd2b4b4e2296d0d93b4c3706cc7c8e26efb2e10c37e9f14d3dfcbf91
3
+ size 247493
videos/63438_0/epoch_20/23_traj_1000023_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7f30c7892fd8fae6b5a053ce5d57ed523e4ef778375333cf6b74730a24c0ea9e
3
+ size 347689
videos/63438_0/epoch_20/24_traj_1000024_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6a4093742e49d9e3a9842c8247289da39103495719d3d5eb002399ad6c2332ab
3
+ size 236268
videos/63438_0/epoch_20/25_traj_1000025_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ef8d60cca3f104b6df05db8325eca2011b55b664732d1d52c0b05b9974b77db9
3
+ size 228655
videos/63438_0/epoch_20/26_traj_1000026_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:770446c564a2aa9e29d0962d5e05138dd59ac141bf194f0da8e11deeb8696fa7
3
+ size 383349
videos/63438_0/epoch_20/27_traj_1000027_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6b5c616f2bc582968932461178152c225aff8232b977de837dbcb459f2aed184
3
+ size 335449
videos/63438_0/epoch_20/28_traj_1000028_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:53a29444b088a98048a9174509f79fbc49e086d58715d63ab5e086bd3e2cc399
3
+ size 315900
videos/63438_0/epoch_20/29_traj_1000029_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5c0f55efe6f499807886b4bd8487d3e298af06c68d9778ddbb52a553ab31408c
3
+ size 371807
videos/63438_0/epoch_20/2_traj_1000002_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:343659c0cf76eb3c70731526483f872e333f9bfec31d49f3acdf75c28bec66e5
3
+ size 386840
videos/63438_0/epoch_20/30_traj_1000030_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a3f0c42682ce40a9272dd1d9e61d52565a3c38c67bbb6b66fa21e55057279280
3
+ size 242906
videos/63438_0/epoch_20/31_traj_1000031_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b134930be32ff9810960170410a8b8cc3ecc2a3ea790db50bb0f5bc530aa585d
3
+ size 333056
videos/63438_0/epoch_20/32_traj_1000032_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:75ab166eae1b96adf9906a9ba9023513924898ee8292e06641eb9a64e5270db9
3
+ size 147677
videos/63438_0/epoch_20/33_traj_1000033_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:edf28cd7ba3c22cb07b5a94b82a076407ca444d1a9d6b23eae26835855cc034f
3
+ size 344607
videos/63438_0/epoch_20/34_traj_1000034_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b67e3234184e9bb6dcd7ada12e8f9b4ccbb533cb54705521738adde745e70fd7
3
+ size 447619
videos/63438_0/epoch_20/35_traj_1000035_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:21724281f2683e23666b5a8a8362ba8a836bade06b80380cee2204bc70064c49
3
+ size 427702
videos/63438_0/epoch_20/36_traj_1000036_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:54f61117ca235a6076177da8a59e839ac55bcfe0fca739f3257ec28bca9ab4a5
3
+ size 240424
videos/63438_0/epoch_20/37_traj_1000037_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cf33289a53bb598873d752a750bcedbd54d16505e5ecd1cf784571119e9d41d9
3
+ size 453776
videos/63438_0/epoch_20/38_traj_1000038_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3decd4f1bb0631d22c4def2c28a719d1de3dcbc65ff1c38a6466b0736503cdb2
3
+ size 278027
videos/63438_0/epoch_20/39_traj_1000039_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b44fe7f2a5955fb340222008de8a6ba3f305bf66ed809525d52b6dc86b21b35a
3
+ size 339146
videos/63438_0/epoch_20/3_traj_1000003_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cb65ccbe340173214996b8b693da190acb1be6218d964964fb422e310236217a
3
+ size 386206
videos/63438_0/epoch_20/40_traj_1000040_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b4a90ac87508a47222f92b455252177b129b32119d5d2fcca82408078f5147f9
3
+ size 344830
videos/63438_0/epoch_20/41_traj_1000041_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:017f04ae2b7ad857cb27b27c8231868a24a40bedd05169aece3d255ddd55dc54
3
+ size 252672
videos/63438_0/epoch_20/42_traj_1000042_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:09d797a502ec55091a23634cf64493cdca87f148990158b6c1bf8db9c7c11916
3
+ size 408710
videos/63438_0/epoch_20/43_traj_1000043_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9a302d92cee33548798c4f4a63e1eb2a6409644915d5931f624ac7025d920ad3
3
+ size 271540
videos/63438_0/epoch_20/44_traj_1000044_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3d981023afc2d6323899f9cdf5ea0c54bd62319e0a474f1ea316a245eaab6c79
3
+ size 420480
videos/63438_0/epoch_20/45_traj_1000045_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d57c3314f8f71a415c5c3a530d7f492505f4f6e5b656733537881a28a7e6b202
3
+ size 364383
videos/63438_0/epoch_20/46_traj_1000046_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ce033b958adbc11048ad2508185f21962699a1d9a766042d207ad40b5adf0c45
3
+ size 364730
videos/63438_0/epoch_20/47_traj_1000047_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a2d64afa407eccf59e73573ed4cb721018524bdef9046ecaad2fa8f95ba1f2e1
3
+ size 423553
videos/63438_0/epoch_20/48_traj_1000048_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:16287832dae8a347b34c82359059868966a4b14fce767452a867e11a4801c090
3
+ size 289365
videos/63438_0/epoch_20/49_traj_1000049_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e4071aa6fb20980a87abb38347a6428f1c48c9e4efcec62ed343a9f994c4c3be
3
+ size 339694
videos/63438_0/epoch_20/4_traj_1000004_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3305fd6206a2a9d99ddd897bb597443d54643998fa371b7b54226ad4267c81e0
3
+ size 430581
videos/63438_0/epoch_20/50_traj_1000050_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1c749bd7c4cfec467c4a35b3ea98c13c4629fdfc93ca67fecf496a71821beb0c
3
+ size 346655
videos/63438_0/epoch_20/51_traj_1000051_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fa66bbedf18cc80a33063d6068f860968e5b3b5db0a539992b417e7ca2cbc609
3
+ size 423061
videos/63438_0/epoch_20/52_traj_1000052_envseed.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:760b071d05b6a12184b8f2e0ba151b40563493650b1b28bec8c94800b4f95af5
3
+ size 388587