serialexperimentsleon commited on
Commit
1c8c569
·
verified ·
1 Parent(s): 3def2d1

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 +144 -0
  2. config.yaml +623 -0
  3. videos/82984_0/epoch_20/0_traj_1000000_envseed.mp4 +3 -0
  4. videos/82984_0/epoch_20/10_traj_1000010_envseed.mp4 +3 -0
  5. videos/82984_0/epoch_20/11_traj_1000011_envseed.mp4 +3 -0
  6. videos/82984_0/epoch_20/12_traj_1000012_envseed.mp4 +3 -0
  7. videos/82984_0/epoch_20/13_traj_1000013_envseed.mp4 +3 -0
  8. videos/82984_0/epoch_20/14_traj_1000014_envseed.mp4 +3 -0
  9. videos/82984_0/epoch_20/15_traj_1000015_envseed.mp4 +3 -0
  10. videos/82984_0/epoch_20/16_traj_1000016_envseed.mp4 +3 -0
  11. videos/82984_0/epoch_20/17_traj_1000017_envseed.mp4 +0 -0
  12. videos/82984_0/epoch_20/18_traj_1000018_envseed.mp4 +3 -0
  13. videos/82984_0/epoch_20/19_traj_1000019_envseed.mp4 +3 -0
  14. videos/82984_0/epoch_20/1_traj_1000001_envseed.mp4 +3 -0
  15. videos/82984_0/epoch_20/20_traj_1000020_envseed.mp4 +3 -0
  16. videos/82984_0/epoch_20/21_traj_1000021_envseed.mp4 +3 -0
  17. videos/82984_0/epoch_20/22_traj_1000022_envseed.mp4 +3 -0
  18. videos/82984_0/epoch_20/23_traj_1000023_envseed.mp4 +3 -0
  19. videos/82984_0/epoch_20/24_traj_1000024_envseed.mp4 +3 -0
  20. videos/82984_0/epoch_20/25_traj_1000025_envseed.mp4 +3 -0
  21. videos/82984_0/epoch_20/26_traj_1000026_envseed.mp4 +3 -0
  22. videos/82984_0/epoch_20/27_traj_1000027_envseed.mp4 +3 -0
  23. videos/82984_0/epoch_20/28_traj_1000028_envseed.mp4 +3 -0
  24. videos/82984_0/epoch_20/29_traj_1000029_envseed.mp4 +3 -0
  25. videos/82984_0/epoch_20/2_traj_1000002_envseed.mp4 +3 -0
  26. videos/82984_0/epoch_20/30_traj_1000030_envseed.mp4 +3 -0
  27. videos/82984_0/epoch_20/31_traj_1000031_envseed.mp4 +3 -0
  28. videos/82984_0/epoch_20/32_traj_1000032_envseed.mp4 +3 -0
  29. videos/82984_0/epoch_20/33_traj_1000033_envseed.mp4 +3 -0
  30. videos/82984_0/epoch_20/34_traj_1000034_envseed.mp4 +3 -0
  31. videos/82984_0/epoch_20/35_traj_1000035_envseed.mp4 +3 -0
  32. videos/82984_0/epoch_20/36_traj_1000036_envseed.mp4 +3 -0
  33. videos/82984_0/epoch_20/37_traj_1000037_envseed.mp4 +3 -0
  34. videos/82984_0/epoch_20/38_traj_1000038_envseed.mp4 +3 -0
  35. videos/82984_0/epoch_20/39_traj_1000039_envseed.mp4 +3 -0
  36. videos/82984_0/epoch_20/3_traj_1000003_envseed.mp4 +0 -0
  37. videos/82984_0/epoch_20/40_traj_1000040_envseed.mp4 +3 -0
  38. videos/82984_0/epoch_20/41_traj_1000041_envseed.mp4 +3 -0
  39. videos/82984_0/epoch_20/42_traj_1000042_envseed.mp4 +3 -0
  40. videos/82984_0/epoch_20/43_traj_1000043_envseed.mp4 +3 -0
  41. videos/82984_0/epoch_20/44_traj_1000044_envseed.mp4 +3 -0
  42. videos/82984_0/epoch_20/45_traj_1000045_envseed.mp4 +3 -0
  43. videos/82984_0/epoch_20/46_traj_1000046_envseed.mp4 +3 -0
  44. videos/82984_0/epoch_20/47_traj_1000047_envseed.mp4 +3 -0
  45. videos/82984_0/epoch_20/48_traj_1000048_envseed.mp4 +3 -0
  46. videos/82984_0/epoch_20/49_traj_1000049_envseed.mp4 +3 -0
  47. videos/82984_0/epoch_20/4_traj_1000004_envseed.mp4 +3 -0
  48. videos/82984_0/epoch_20/50_traj_1000050_envseed.mp4 +0 -0
  49. videos/82984_0/epoch_20/51_traj_1000051_envseed.mp4 +3 -0
  50. videos/82984_0/epoch_20/52_traj_1000052_envseed.mp4 +3 -0
.gitattributes CHANGED
@@ -33,3 +33,147 @@ 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/82984_0/epoch_20/39_traj_1000039_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
37
+ videos/82984_0/epoch_20/1_traj_1000001_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
38
+ videos/82984_0/epoch_20/67_traj_1000067_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
39
+ videos/82984_0/epoch_20/99_traj_1000099_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
40
+ videos/82984_0/epoch_20/28_traj_1000028_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
41
+ videos/82984_0/epoch_20/97_traj_1000097_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
42
+ videos/82984_0/epoch_20/35_traj_1000035_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
43
+ videos/82984_0/epoch_20/19_traj_1000019_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
44
+ videos/82984_0/epoch_20/9_traj_1000009_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
45
+ videos/82984_0/epoch_20/24_traj_1000024_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
46
+ videos/82984_0/epoch_20/81_traj_1000081_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
47
+ videos/82984_0/epoch_20/37_traj_1000037_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
48
+ videos/82984_0/epoch_20/93_traj_1000093_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
49
+ videos/82984_0/epoch_20/51_traj_1000051_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
50
+ videos/82984_0/epoch_20/73_traj_1000073_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
51
+ videos/82984_0/epoch_20/64_traj_1000064_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
52
+ videos/82984_0/epoch_20/53_traj_1000053_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
53
+ videos/82984_0/epoch_20/8_traj_1000008_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
54
+ videos/82984_0/epoch_20/46_traj_1000046_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
55
+ videos/82984_0/epoch_20/44_traj_1000044_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
56
+ videos/82984_0/epoch_20/95_traj_1000095_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
57
+ videos/82984_0/epoch_20/10_traj_1000010_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
58
+ videos/82984_0/epoch_20/49_traj_1000049_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
59
+ videos/82984_0/epoch_20/74_traj_1000074_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
60
+ videos/82984_0/epoch_20/11_traj_1000011_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
61
+ videos/82984_0/epoch_20/98_traj_1000098_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
62
+ videos/82984_0/epoch_20/14_traj_1000014_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
63
+ videos/82984_0/epoch_20/20_traj_1000020_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
64
+ videos/82984_0/epoch_20/55_traj_1000055_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
65
+ videos/82984_0/epoch_20/65_traj_1000065_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
66
+ videos/82984_0/epoch_20/48_traj_1000048_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
67
+ videos/82984_0/epoch_20/0_traj_1000000_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
68
+ videos/82984_0/epoch_20/68_traj_1000068_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
69
+ videos/82984_0/epoch_20/15_traj_1000015_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
70
+ videos/82984_0/epoch_20/43_traj_1000043_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
71
+ videos/82984_0/epoch_20/75_traj_1000075_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
72
+ videos/82984_0/epoch_20/89_traj_1000089_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
73
+ videos/82984_0/epoch_20/79_traj_1000079_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
74
+ videos/82984_0/epoch_20/62_traj_1000062_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
75
+ videos/82984_0/epoch_20/2_traj_1000002_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
76
+ videos/82984_0/epoch_20/57_traj_1000057_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
77
+ videos/82984_0/epoch_20/92_traj_1000092_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
78
+ videos/82984_0/epoch_20/29_traj_1000029_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
79
+ videos/82984_0/epoch_20/22_traj_1000022_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
80
+ videos/82984_0/epoch_20/33_traj_1000033_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
81
+ videos/82984_0/epoch_20/80_traj_1000080_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
82
+ videos/82984_0/epoch_20/30_traj_1000030_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
83
+ videos/82984_0/epoch_20/84_traj_1000084_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
84
+ videos/82984_0/epoch_20/70_traj_1000070_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
85
+ videos/82984_0/epoch_20/7_traj_1000007_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
86
+ videos/82984_0/epoch_20/60_traj_1000060_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
87
+ videos/82984_0/epoch_20/21_traj_1000021_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
88
+ videos/82984_0/epoch_20/82_traj_1000082_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
89
+ videos/82984_0/epoch_20/69_traj_1000069_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
90
+ videos/82984_0/epoch_20/36_traj_1000036_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
91
+ videos/82984_0/epoch_20/90_traj_1000090_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
92
+ videos/82984_0/epoch_20/61_traj_1000061_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
93
+ videos/82984_0/epoch_20/63_traj_1000063_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
94
+ videos/82984_0/epoch_20/94_traj_1000094_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
95
+ videos/82984_0/epoch_20/59_traj_1000059_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
96
+ videos/82984_0/epoch_20/87_traj_1000087_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
97
+ videos/82984_0/epoch_20/45_traj_1000045_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
98
+ videos/82984_0/epoch_20/31_traj_1000031_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
99
+ videos/82984_0/epoch_20/66_traj_1000066_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
100
+ videos/82984_0/epoch_20/83_traj_1000083_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
101
+ videos/82984_0/epoch_20/41_traj_1000041_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
102
+ videos/82984_0/epoch_20/16_traj_1000016_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
103
+ videos/82984_0/epoch_20/96_traj_1000096_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
104
+ videos/82984_0/epoch_20/18_traj_1000018_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
105
+ videos/82984_0/epoch_20/88_traj_1000088_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
106
+ videos/82984_0/epoch_20/40_traj_1000040_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
107
+ videos/82984_0/epoch_20/85_traj_1000085_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
108
+ videos/82984_0/epoch_20/52_traj_1000052_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
109
+ videos/82984_0/epoch_20/72_traj_1000072_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
110
+ videos/82984_0/epoch_20/34_traj_1000034_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
111
+ videos/82984_0/epoch_20/12_traj_1000012_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
112
+ videos/82984_0/epoch_20/13_traj_1000013_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
113
+ videos/82984_0/epoch_20/23_traj_1000023_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
114
+ videos/82984_0/epoch_20/38_traj_1000038_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
115
+ videos/82984_0/epoch_20/5_traj_1000005_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
116
+ videos/82984_0/epoch_20/86_traj_1000086_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
117
+ videos/82984_0/epoch_20/77_traj_1000077_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
118
+ videos/82984_0/epoch_20/78_traj_1000078_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
119
+ videos/82984_0/epoch_20/76_traj_1000076_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
120
+ videos/82984_0/epoch_20/27_traj_1000027_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
121
+ videos/82984_0/epoch_20/26_traj_1000026_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
122
+ videos/82984_0/epoch_20/56_traj_1000056_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
123
+ videos/82984_0/epoch_20/25_traj_1000025_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
124
+ videos/82984_0/epoch_20/71_traj_1000071_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
125
+ videos/82984_0/epoch_20/4_traj_1000004_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
126
+ videos/82984_0/epoch_20/32_traj_1000032_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
127
+ videos/82984_0/epoch_20/42_traj_1000042_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
128
+ videos/82984_0/epoch_20/47_traj_1000047_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
129
+ videos/82984_0/epoch_20/58_traj_1000058_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
130
+ videos/82984_0/epoch_20/6_traj_1000006_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
131
+ videos/82984_0/epoch_80/392_traj_1000092_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
132
+ videos/82984_0/epoch_80/351_traj_1000051_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
133
+ videos/82984_0/epoch_80/345_traj_1000045_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
134
+ videos/82984_0/epoch_80/332_traj_1000032_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
135
+ videos/82984_0/epoch_80/346_traj_1000046_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
136
+ videos/82984_0/epoch_80/359_traj_1000059_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
137
+ videos/82984_0/epoch_80/347_traj_1000047_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
138
+ videos/82984_0/epoch_80/307_traj_1000007_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
139
+ videos/82984_0/epoch_80/308_traj_1000008_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
140
+ videos/82984_0/epoch_80/388_traj_1000088_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
141
+ videos/82984_0/epoch_80/330_traj_1000030_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
142
+ videos/82984_0/epoch_80/343_traj_1000043_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
143
+ videos/82984_0/epoch_80/372_traj_1000072_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
144
+ videos/82984_0/epoch_80/357_traj_1000057_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
145
+ videos/82984_0/epoch_80/350_traj_1000050_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
146
+ videos/82984_0/epoch_80/370_traj_1000070_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
147
+ videos/82984_0/epoch_80/399_traj_1000099_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
148
+ videos/82984_0/epoch_80/336_traj_1000036_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
149
+ videos/82984_0/epoch_80/379_traj_1000079_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
150
+ videos/82984_0/epoch_80/375_traj_1000075_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
151
+ videos/82984_0/epoch_80/348_traj_1000048_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
152
+ videos/82984_0/epoch_80/327_traj_1000027_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
153
+ videos/82984_0/epoch_80/356_traj_1000056_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
154
+ videos/82984_0/epoch_80/325_traj_1000025_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
155
+ videos/82984_0/epoch_80/302_traj_1000002_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
156
+ videos/82984_0/epoch_80/337_traj_1000037_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
157
+ videos/82984_0/epoch_80/333_traj_1000033_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
158
+ videos/82984_0/epoch_80/361_traj_1000061_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
159
+ videos/82984_0/epoch_80/371_traj_1000071_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
160
+ videos/82984_0/epoch_80/305_traj_1000005_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
161
+ videos/82984_0/epoch_80/369_traj_1000069_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
162
+ videos/82984_0/epoch_80/319_traj_1000019_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
163
+ videos/82984_0/epoch_80/349_traj_1000049_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
164
+ videos/82984_0/epoch_80/301_traj_1000001_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
165
+ videos/82984_0/epoch_80/378_traj_1000078_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
166
+ videos/82984_0/epoch_80/360_traj_1000060_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
167
+ videos/82984_0/epoch_80/384_traj_1000084_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
168
+ videos/82984_0/epoch_80/326_traj_1000026_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
169
+ videos/82984_0/epoch_80/398_traj_1000098_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
170
+ videos/82984_0/epoch_80/322_traj_1000022_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
171
+ videos/82984_0/epoch_80/382_traj_1000082_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
172
+ videos/82984_0/epoch_80/363_traj_1000063_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
173
+ videos/82984_0/epoch_80/331_traj_1000031_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
174
+ videos/82984_0/epoch_80/366_traj_1000066_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
175
+ videos/82984_0/epoch_80/310_traj_1000010_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
176
+ videos/82984_0/epoch_80/354_traj_1000054_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
177
+ videos/82984_0/epoch_80/312_traj_1000012_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
178
+ videos/82984_0/epoch_80/324_traj_1000024_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
179
+ videos/82984_0/epoch_80/385_traj_1000085_envseed.mp4 filter=lfs diff=lfs merge=lfs -text
config.yaml ADDED
@@ -0,0 +1,623 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: '82984_0'
16
+ wandb_notes: '82984_0_'
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
+ grasped_book_config:
77
+ randomize_color: false
78
+ randomize_density: false
79
+ density_randomization_bounds:
80
+ - 650
81
+ - 850
82
+ randomize_height: false
83
+ height_randomization_bounds:
84
+ - 0.165
85
+ - 0.25
86
+ randomize_width: false
87
+ width_randomization_bounds:
88
+ - 0.03
89
+ - 0.065
90
+ randomize_length: false
91
+ length_randomization_bounds:
92
+ - 0.1
93
+ - 0.15
94
+ env_books_config:
95
+ num_env_books: 8
96
+ randomize_color: false
97
+ randomize_density: false
98
+ density_randomization_bounds:
99
+ - 655
100
+ - 1015
101
+ randomize_height: false
102
+ height_randomization_bounds:
103
+ - 0.2475
104
+ - 0.2525
105
+ randomize_width: false
106
+ width_randomization_bounds:
107
+ - 0.015
108
+ - 0.05
109
+ randomize_length: false
110
+ length_randomization_bounds:
111
+ - 0.15
112
+ - 0.2
113
+ shuffle_mode: none
114
+ slot_config:
115
+ negative_tolerance: 0.0035
116
+ left_of_book_index: 4
117
+ y_randomization_bounds:
118
+ - -0.05
119
+ - 0.05
120
+ book_ends_config:
121
+ mode: spring
122
+ length: 0.2
123
+ width: 0.025
124
+ height: 0.25
125
+ mass: 1.0
126
+ color: '#808080'
127
+ friction: 0.0
128
+ wall_height: 0.25
129
+ wall_length: 0.2
130
+ wall_width: 0.025
131
+ travel_limit: 0.125
132
+ joint_stiffness: 100.0
133
+ joint_damping: 20.0
134
+ record_sim_videos: true
135
+ resume_wandb_run: true
136
+ train_demo_idxs_list_or_num: -1
137
+ log_train_every_steps: 25
138
+ name_of_expert_demo: 519_sim_demos_w_recovery_leftof4thbook_springbookends_nograspedrand_noenvrand_slotrand_20hz_act
139
+ expert_dataset_dirpath: ${root_dir}/FISH/expert_demos/${suite.name}/${task_name}/${name_of_expert_demo}
140
+ store_dataset_in_memory: false
141
+ expert_dataset: ${expert_dataset_dirpath}/demos.zarr
142
+ action_key: action
143
+ semantic_demo_grouping_name: semantic_demo_grouping.yaml
144
+ semantic_demo_grouping: ${expert_dataset_dirpath}/${semantic_demo_grouping_name}
145
+ include_groups_list: all
146
+ expert_dataset_config: ${expert_dataset_dirpath}/demo_config.yaml
147
+ name_of_valid_demo: 519_sim_demos_w_recovery_leftof4thbook_springbookends_nograspedrand_noenvrand_slotrand_20hz_act
148
+ train_num_groups: 489
149
+ valid_dataset_dir: ${root_dir}/FISH/expert_demos/${suite.name}/${task_name}/${name_of_valid_demo}/demos.zarr
150
+ valid_demo_idxs_list_or_num: null
151
+ val_num_groups: 30
152
+ universal_unseen_env_seed_start: 1000000
153
+ load_bc: ${agent.load_checkpoint}
154
+ checkpoint_epoch_list:
155
+ - 99
156
+ - 199
157
+ - 299
158
+ - 399
159
+ - 499
160
+ - 599
161
+ - 699
162
+ - 799
163
+ - 899
164
+ - 999
165
+ - 1249
166
+ - 1499
167
+ - 1749
168
+ - 1999
169
+ - 2999
170
+ - 3999
171
+ - 4999
172
+ - 5999
173
+ - 6999
174
+ - 7999
175
+ - 8999
176
+ - 9999
177
+ snapshot_root_dir: /mnt/grasp_high_usage/leonmkim/contact_estimation/FISH
178
+ save_snapshot: true
179
+ save_last_snapshot: true
180
+ save_snapshot_when_done: true
181
+ top_k_checkpoints: 10
182
+ save_snapshot_link_to_weights_dir: deprecated
183
+ restore_missing_checkpoint: null
184
+ restore_and_eval_all_missing_checkpoints: false
185
+ bc_regularize: false
186
+ bc_weight_type: qfilter
187
+ username: leonmkim
188
+ hostname: grasp-login1
189
+ local_snapshot_root_dir: /mnt/bighdd/fish_contact_backup
190
+ local_snapshot_dir: ${local_snapshot_root_dir}/exp_local/${suite.name}_${obs_type}/${task_name}
191
+ experiment_dir: ./exp_local/${suite.name}_${obs_type}/${task_name}/${wandb_run_id}
192
+ agent:
193
+ _target_: agent.diffusion_policy.DiffusionPolicyAgent
194
+ name: diffusion_policy
195
+ load_checkpoint: ${eval}
196
+ device: ${device}
197
+ n_obs_steps: ${.config.policy_cfg.n_obs_steps}
198
+ suite_name: ${suite.name}
199
+ obs_type: ${obs_type}
200
+ enable_arm: ${eval}
201
+ enable_camera: ${eval}
202
+ use_tb: ${use_tb}
203
+ desired_image_shape:
204
+ - 13
205
+ - 180
206
+ - 240
207
+ orig_cam_shape:
208
+ - 3
209
+ - 240
210
+ - 320
211
+ actions_prenormalized: ${simulation}
212
+ config:
213
+ _target_: agent.diffusion_policy.DiffusionPolicyAgentConfig
214
+ compile: false
215
+ device: ${device}
216
+ cam_resize_shape: ${agent.desired_image_shape}
217
+ orig_cam_shape: ${agent.orig_cam_shape}
218
+ policy_frequency: ${policy_frequency}
219
+ interpolation_frequency: ${interpolation_frequency}
220
+ deterministic_actions: true
221
+ deterministic_actions_seed: 0
222
+ policy_cfg:
223
+ _target_: lerobot.common.policies.diffusion.configuration_diffusion.DiffusionConfig
224
+ n_obs_steps: 1
225
+ horizon: 28
226
+ n_action_steps: ${agent.config.policy_cfg.horizon}
227
+ output_shapes:
228
+ action:
229
+ - 7
230
+ input_normalization_modes:
231
+ observation.image: mean_std
232
+ observation.state: min_max
233
+ observation.action_history: min_max
234
+ output_normalization_modes:
235
+ action: min_max
236
+ vision_backbone: resnet18
237
+ crop_distractors_tlhw: null
238
+ pretrained_backbone_weights: null
239
+ transforms:
240
+ - _target_: torchvision.transforms.v2.Grayscale
241
+ num_output_channels: 3
242
+ use_group_norm: true
243
+ spatial_softmax_num_keypoints: 32
244
+ action_history_encoder_config:
245
+ _target_: lerobot.common.policies.diffusion.configuration_diffusion.Unet1dEncoderConfig
246
+ in_channels: 7
247
+ out_channels: 32
248
+ history_length: 4
249
+ kernel_size: ${agent.config.policy_cfg.kernel_size}
250
+ downsample_kernel_size: 3
251
+ downsample_stride: 2
252
+ downsample_padding: 1
253
+ down_dims:
254
+ - 256
255
+ - 512
256
+ - 1024
257
+ kernel_size: 5
258
+ n_groups: 8
259
+ diffusion_step_embed_dim: 128
260
+ use_film_scale_modulation: true
261
+ noise_scheduler_type: DDIM
262
+ beta_schedule: squaredcos_cap_v2
263
+ beta_start: 0.0001
264
+ beta_end: 0.02
265
+ prediction_type: epsilon
266
+ clip_sample: true
267
+ clip_sample_range: 1.0
268
+ num_train_timesteps: 50
269
+ num_inference_steps: 10
270
+ do_mask_loss_for_padding: false
271
+ start_horizon_at_current_step: true
272
+ input_shapes:
273
+ observation.image:
274
+ - 13
275
+ - 180
276
+ - 240
277
+ context_observation.image:
278
+ - 13
279
+ - 180
280
+ - 240
281
+ observation.state:
282
+ - 8
283
+ train_cfg:
284
+ _target_: utils.TrainConfig
285
+ lr: 0.0001
286
+ lr_scheduler: cosine
287
+ lr_warmup_steps: 500
288
+ adam_betas:
289
+ - 0.95
290
+ - 0.999
291
+ adam_eps: 1.0e-08
292
+ adam_weight_decay: 1.0e-06
293
+ grad_clip_norm: 10
294
+ offline_steps: 533000
295
+ use_amp: true
296
+ optimizer: adamw
297
+ observation_cfg:
298
+ _target_: agent.encoder.VisualFeatureSet
299
+ use_depth: ${use_depth}
300
+ use_color: ${use_color}
301
+ mask_input_dict:
302
+ _target_: agent.encoder.MaskInputDict
303
+ enable: ${use_masks}
304
+ representation: ${mask_representation}
305
+ mask_list: ${mask_list}
306
+ segmentation_model_name: ${segmentation_model_name}
307
+ crop_input_config:
308
+ _target_: agent.encoder.CropInputConfig
309
+ color_crop_type: ${color_crop_type}
310
+ depth_crop_type: ${depth_crop_type}
311
+ segmask_crop_type: ${segmask_crop_type}
312
+ crop_hw: ${crop_hw}
313
+ crop_down_offset: ${crop_down_offset}
314
+ add_crop_binary_mask: ${add_crop_binary_mask}
315
+ add_coord_conv_map: ${add_coord_conv_map}
316
+ context_input_config:
317
+ _target_: agent.encoder.ContextInputConfig
318
+ use_color: ${use_context_color}
319
+ use_depth: ${use_context_depth}
320
+ mask_input_dict:
321
+ _target_: agent.encoder.MaskInputDict
322
+ enable: ${use_context_segmask}
323
+ representation: ${mask_representation}
324
+ mask_list: ${mask_list}
325
+ segmentation_model_name: ${segmentation_model_name}
326
+ crop_input_config:
327
+ _target_: agent.encoder.CropInputConfig
328
+ color_crop_type: ${context_color_crop_type}
329
+ depth_crop_type: ${context_depth_crop_type}
330
+ segmask_crop_type: ${context_segmask_crop_type}
331
+ crop_hw: ${crop_hw}
332
+ crop_down_offset: ${crop_down_offset}
333
+ add_crop_binary_mask: ${context_add_crop_binary_mask}
334
+ add_coord_conv_map: ${context_add_coord_conv_map}
335
+ mask_soft_approx_scheduler_config:
336
+ _target_: agent.encoder.MaskSoftApproxSchedulerConfig
337
+ num_steps: 40000
338
+ initial_value: 10.0
339
+ final_value: 1000.0
340
+ interpolation_scheme: cosine
341
+ use_contact_map: ${use_contact_map}
342
+ use_sdf_maps: ${use_sdf_maps}
343
+ use_normals_maps: ${use_normals_maps}
344
+ which_objects: ${which_objects}
345
+ grasped_dtc_max_value: ${grasped_dtc_max_value}
346
+ env_dtc_max_value: ${env_dtc_max_value}
347
+ grasped_normals_mask_max_dtc_value: ${grasped_normals_mask_max_dtc_value}
348
+ env_normals_mask_max_dtc_value: ${env_normals_mask_max_dtc_value}
349
+ clamp_dtc: ${clamp_dtc}
350
+ max_contact_prob: ${max_contact_prob}
351
+ mask_normals_within_sdf: ${mask_normals_within_sdf}
352
+ dtc_adaptive_normalization: ${dtc_adaptive_normalization}
353
+ adaptive_normals_mask: ${adaptive_normals_mask}
354
+ max_depth: ${max_depth}
355
+ image_shape: ${agent.desired_image_shape}
356
+ learnable_contact_preprocess_params: ${learnable_contact_preprocess_params}
357
+ learning_rate: 0.0001
358
+ weight_decay: 0.0
359
+ contact_model_name: ${contact_model_name}
360
+ zero_centered: false
361
+ crop_distractors_tlhw: ${agent.config.policy_cfg.crop_distractors_tlhw}
362
+ suite:
363
+ suite: frankagym
364
+ name: frankagym
365
+ frame_stack: ${agent.n_obs_steps}
366
+ action_repeat: 1
367
+ discount: 0.99
368
+ hidden_dim: 1024
369
+ num_train_frames: 1000000
370
+ num_seed_frames: 0
371
+ num_train_epochs: 1000
372
+ validate_every_epochs: 20
373
+ eval_on_sim_every_epochs: 20
374
+ validate_diffusion_on_action_loss_every_epochs: 20
375
+ eval_on_sim_every_epochs_diffusion: 500
376
+ train_eval_diffusion_on_action_loss_every_epochs: 20
377
+ check_topk_every_epochs: 20
378
+ save_snapshot_every_epochs: 25
379
+ num_seen_eval_envs: 0
380
+ num_unseen_eval_envs: 100
381
+ evals_per_env: 1
382
+ eval_every_frames: 2000
383
+ num_eval_episodes: 5
384
+ save_snapshot: true
385
+ wait_for_user_to_start_episode: true
386
+ task_make_fn:
387
+ _target_: suite.frankagym.make
388
+ name: ${task_name}
389
+ height: 240
390
+ width: 320
391
+ frame_stack: ${suite.frame_stack}
392
+ action_repeat: ${suite.action_repeat}
393
+ seed: ${seed}
394
+ enable_arm: ${agent.enable_arm}
395
+ enable_gripper: ${enable_gripper}
396
+ start_with_gripper_open: ${start_with_gripper_open}
397
+ enable_camera: ${agent.enable_camera}
398
+ path_to_depth_extrinsics: ${path_to_depth_extrinsics}
399
+ contact_estimation_model_ckpt_path: ${contact_estimation_model_ckpt_path}
400
+ x_limit: ${x_limit}
401
+ y_limit: ${y_limit}
402
+ z_limit: ${z_limit}
403
+ device: ${device}
404
+ interpolation_frequency: ${interpolation_frequency}
405
+ policy_frequency: ${policy_frequency}
406
+ debug_timestamps: ${debug_timestamps}
407
+ stop_after_action: ${stop_after_action}
408
+ open_loop: ${open_loop}
409
+ wait_for_new_camera_frames: ${wait_for_new_camera_frames}
410
+ action_key: ${action_key}
411
+ action_trajectory_horizon: ${agent.config.policy_cfg.horizon}
412
+ action_trajectories: ${action_trajectories}
413
+ path_to_zarr_dataset: ${expert_dataset}
414
+ agent_policy_cfg:
415
+ _target_: agent.diffusion_policy.DiffusionPolicyAgentConfig
416
+ compile: false
417
+ device: ${device}
418
+ cam_resize_shape: ${agent.desired_image_shape}
419
+ orig_cam_shape: ${agent.orig_cam_shape}
420
+ policy_frequency: ${policy_frequency}
421
+ interpolation_frequency: ${interpolation_frequency}
422
+ deterministic_actions: true
423
+ deterministic_actions_seed: 0
424
+ policy_cfg:
425
+ _target_: lerobot.common.policies.diffusion.configuration_diffusion.DiffusionConfig
426
+ n_obs_steps: 1
427
+ horizon: 28
428
+ n_action_steps: ${agent.config.policy_cfg.horizon}
429
+ output_shapes:
430
+ action:
431
+ - 7
432
+ input_normalization_modes:
433
+ observation.image: mean_std
434
+ observation.state: min_max
435
+ observation.action_history: min_max
436
+ output_normalization_modes:
437
+ action: min_max
438
+ vision_backbone: resnet18
439
+ crop_distractors_tlhw: null
440
+ pretrained_backbone_weights: null
441
+ transforms:
442
+ - _target_: torchvision.transforms.v2.Grayscale
443
+ num_output_channels: 3
444
+ use_group_norm: true
445
+ spatial_softmax_num_keypoints: 32
446
+ action_history_encoder_config:
447
+ _target_: lerobot.common.policies.diffusion.configuration_diffusion.Unet1dEncoderConfig
448
+ in_channels: 7
449
+ out_channels: 32
450
+ history_length: 4
451
+ kernel_size: ${agent.config.policy_cfg.kernel_size}
452
+ downsample_kernel_size: 3
453
+ downsample_stride: 2
454
+ downsample_padding: 1
455
+ down_dims:
456
+ - 256
457
+ - 512
458
+ - 1024
459
+ kernel_size: 5
460
+ n_groups: 8
461
+ diffusion_step_embed_dim: 128
462
+ use_film_scale_modulation: true
463
+ noise_scheduler_type: DDIM
464
+ beta_schedule: squaredcos_cap_v2
465
+ beta_start: 0.0001
466
+ beta_end: 0.02
467
+ prediction_type: epsilon
468
+ clip_sample: true
469
+ clip_sample_range: 1.0
470
+ num_train_timesteps: 50
471
+ num_inference_steps: 10
472
+ do_mask_loss_for_padding: false
473
+ start_horizon_at_current_step: true
474
+ input_shapes:
475
+ observation.image:
476
+ - 13
477
+ - 180
478
+ - 240
479
+ context_observation.image:
480
+ - 13
481
+ - 180
482
+ - 240
483
+ observation.state:
484
+ - 8
485
+ train_cfg:
486
+ _target_: utils.TrainConfig
487
+ lr: 0.0001
488
+ lr_scheduler: cosine
489
+ lr_warmup_steps: 500
490
+ adam_betas:
491
+ - 0.95
492
+ - 0.999
493
+ adam_eps: 1.0e-08
494
+ adam_weight_decay: 1.0e-06
495
+ grad_clip_norm: 10
496
+ offline_steps: ${num_train_frames_diffusion}
497
+ use_amp: true
498
+ optimizer: adamw
499
+ observation_cfg:
500
+ _target_: agent.encoder.VisualFeatureSet
501
+ use_depth: ${use_depth}
502
+ use_color: ${use_color}
503
+ mask_input_dict:
504
+ _target_: agent.encoder.MaskInputDict
505
+ enable: ${use_masks}
506
+ representation: ${mask_representation}
507
+ mask_list: ${mask_list}
508
+ segmentation_model_name: ${segmentation_model_name}
509
+ crop_input_config:
510
+ _target_: agent.encoder.CropInputConfig
511
+ color_crop_type: ${color_crop_type}
512
+ depth_crop_type: ${depth_crop_type}
513
+ segmask_crop_type: ${segmask_crop_type}
514
+ crop_hw: ${crop_hw}
515
+ crop_down_offset: ${crop_down_offset}
516
+ add_crop_binary_mask: ${add_crop_binary_mask}
517
+ add_coord_conv_map: ${add_coord_conv_map}
518
+ context_input_config:
519
+ _target_: agent.encoder.ContextInputConfig
520
+ use_color: ${use_context_color}
521
+ use_depth: ${use_context_depth}
522
+ mask_input_dict:
523
+ _target_: agent.encoder.MaskInputDict
524
+ enable: ${use_context_segmask}
525
+ representation: ${mask_representation}
526
+ mask_list: ${mask_list}
527
+ segmentation_model_name: ${segmentation_model_name}
528
+ crop_input_config:
529
+ _target_: agent.encoder.CropInputConfig
530
+ color_crop_type: ${context_color_crop_type}
531
+ depth_crop_type: ${context_depth_crop_type}
532
+ segmask_crop_type: ${context_segmask_crop_type}
533
+ crop_hw: ${crop_hw}
534
+ crop_down_offset: ${crop_down_offset}
535
+ add_crop_binary_mask: ${context_add_crop_binary_mask}
536
+ add_coord_conv_map: ${context_add_coord_conv_map}
537
+ mask_soft_approx_scheduler_config:
538
+ _target_: agent.encoder.MaskSoftApproxSchedulerConfig
539
+ num_steps: 40000
540
+ initial_value: 10.0
541
+ final_value: 1000.0
542
+ interpolation_scheme: cosine
543
+ use_contact_map: ${use_contact_map}
544
+ use_sdf_maps: ${use_sdf_maps}
545
+ use_normals_maps: ${use_normals_maps}
546
+ which_objects: ${which_objects}
547
+ grasped_dtc_max_value: ${grasped_dtc_max_value}
548
+ env_dtc_max_value: ${env_dtc_max_value}
549
+ grasped_normals_mask_max_dtc_value: ${grasped_normals_mask_max_dtc_value}
550
+ env_normals_mask_max_dtc_value: ${env_normals_mask_max_dtc_value}
551
+ clamp_dtc: ${clamp_dtc}
552
+ max_contact_prob: ${max_contact_prob}
553
+ mask_normals_within_sdf: ${mask_normals_within_sdf}
554
+ dtc_adaptive_normalization: ${dtc_adaptive_normalization}
555
+ adaptive_normals_mask: ${adaptive_normals_mask}
556
+ max_depth: ${max_depth}
557
+ image_shape: ${agent.desired_image_shape}
558
+ learnable_contact_preprocess_params: ${learnable_contact_preprocess_params}
559
+ learning_rate: 0.0001
560
+ weight_decay: 0.0
561
+ contact_model_name: ${contact_model_name}
562
+ zero_centered: false
563
+ crop_distractors_tlhw: ${agent.config.policy_cfg.crop_distractors_tlhw}
564
+ true_action_history: ${true_action_history}
565
+ num_train_frames_bc: 50000
566
+ num_train_frames_drq: 1100000
567
+ stddev_schedule_drq: linear(1.0,0.1,100000)
568
+ task_name: FrankaInsertion-v1
569
+ num_train_frames_vinn: 25000
570
+ num_train_frames_diffusion: 1000000
571
+ num_train_epochs_bc: 5000
572
+ num_train_epochs_diffusion: 1000
573
+ validate_every_epochs_bc: 5
574
+ validate_every_epochs_diffusion: 20
575
+ validate_diffusion_on_action_loss_every_epochs: 20
576
+ train_eval_diffusion_on_action_loss_every_epochs: 20
577
+ eval_on_sim_every_epochs_diffusion: 20
578
+ check_topk_every_epochs: 5
579
+ check_topk_every_epochs_diffusion: 20
580
+ save_snapshot_every_epochs_diffusion: 25
581
+ x_limit:
582
+ - 0.2
583
+ - 0.7
584
+ y_limit:
585
+ - -0.4
586
+ - 0.4
587
+ z_limit:
588
+ - -0.05
589
+ - 0.55
590
+ home_displacement:
591
+ - 0.55
592
+ - 0.0
593
+ - 0.55
594
+ - 180.0
595
+ - 0.0
596
+ - 0.0
597
+ enable_gripper: true
598
+ start_with_gripper_open: true
599
+ offset_mask:
600
+ - 1
601
+ - 1
602
+ - 1
603
+ - 1
604
+ - 1
605
+ - 1
606
+ path_to_depth_extrinsics: ~/fish_leon/FISH/cfgs/camera_poses/camera_poses_L515/20240904-122305/color_tf_world.npy
607
+ test:
608
+ username: leonmkim
609
+ hostname: grasp-login1
610
+ name_of_expert_demo: 112_240x320_all_twodim_left_to_right_annotated_start_idx_5hz_zstd7_EE_pxl_coords_expert_demos_imp_act
611
+ root_dir: ~/fish_leon
612
+ expert_dataset_dirpath: ${test.root_dir}/FISH/expert_demos/${suite.name}/${task_name}/${test.name_of_expert_demo}
613
+ expert_dataset: ${test.expert_dataset_dirpath}/demos.zarr
614
+ semantic_demo_grouping_name: semantic_demo_grouping.yaml
615
+ semantic_demo_grouping: ${test.expert_dataset_dirpath}/${test.semantic_demo_grouping_name}
616
+ include_groups_list: all
617
+ batch_size: 128
618
+ store_dataset_in_memory: false
619
+ use_tb: true
620
+ local_snapshot_root_dir: /mnt/bighdd/fish_contact_backup
621
+ local_snapshot_dir: ${test.local_snapshot_root_dir}/exp_local/${suite.name}_${obs_type}/${task_name}
622
+ resume_wandb_run: false
623
+ feature_type: 180x240_1_RGB_D_2.0
videos/82984_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:195319938ce6bea51d9d03554cfbe452f8b20e79fc61b46a4ce5c4503441893a
3
+ size 154757
videos/82984_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:2346da87b7a9ab9073a74e569b5f374cdf14bb6287d37ab0e4522c1a7017b897
3
+ size 156193
videos/82984_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:3bd8320bed52de74dd00235196acea88ff3a5c45c24b9a4f8d075a11cf907536
3
+ size 151561
videos/82984_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:96299f7402789118ec17573895a076e5e7f4d60b910f7ab8787980b317381ce4
3
+ size 154657
videos/82984_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:2b36ec9ccbc84c99a55b7ea2acd2320c649455edaab7ccf167a1fc2e3373963f
3
+ size 157108
videos/82984_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:091a659ea00e56e66b5e24659bb094de7b85ec026fb7b9cc62bc6efa4a145ded
3
+ size 154029
videos/82984_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:679c80817d2bf511e6ea83de965acdf10858d825a1d26c39620d74c230e6a63d
3
+ size 154043
videos/82984_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:17331754ebd3f540b81324c9369e9dd8a25d7f97b5cac58684b7421e7bd2e9f4
3
+ size 155825
videos/82984_0/epoch_20/17_traj_1000017_envseed.mp4 ADDED
Binary file (84 kB). View file
 
videos/82984_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:a3dd2dea3a3e21bdaf5965ddaa4b309239f082bff3acd8e631114f3c5ecaf83c
3
+ size 154174
videos/82984_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:bdca2347aec75e8056d0b86c67071da6dcd748a70cc65ef905cd94d78612f4e5
3
+ size 153169
videos/82984_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:7dc49b1122dd12bdd012f9f66ba875041443b6602c30841bdcf48603d517446d
3
+ size 154303
videos/82984_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:b400e5c73fcf868c6cac8c918359d0453469b702e6bad6df9b2d60b2a1061f1d
3
+ size 151274
videos/82984_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:f5c761149773fbe6a9d2a028415ec971b077d365d900b8eb2775c279fa7ba67f
3
+ size 154642
videos/82984_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:dea9f1c14411cb822fa79deb2c90d8375838e59662bf2fe36946b2e6daf846ce
3
+ size 152356
videos/82984_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:67e8bff19cfc6e0c83515bdd380ef442548231039908c3565f0cee9e70fd8a20
3
+ size 153995
videos/82984_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:4d015f04d13cf65cf7eed53763ea152ea4b25ded5bd71abcc6bf05a42fedfa90
3
+ size 158096
videos/82984_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:f7abc2ef6788f5c994faca2b63509feb61156e50e6f4d8bb59dc4ed99a607c4a
3
+ size 157303
videos/82984_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:407590a08f2d4deadb487ebc077025a9cf97b9e1b8342f5fffe4b15ab205fbe1
3
+ size 158268
videos/82984_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:938f24012b9d1c02bbd7e64340a5edf51e16f1fe25c7bee6139f9c49efec50ab
3
+ size 152057
videos/82984_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:b261ac053349e9cf040073ff77b073bc435260a20bd03072d86ce68561e6e881
3
+ size 156599
videos/82984_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:6968236814f3746d11f6d21048c38107b273af614ab923406951d2044870f03a
3
+ size 151785
videos/82984_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:21b43b642f370801250019123d76afeadafbd78f8c753113fd3a2f4c39f2465b
3
+ size 163773
videos/82984_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:f7f2b5fa0960a26f6ce3e67cb0dd135e1862a6eca570de741960400ef22205a0
3
+ size 153522
videos/82984_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:7f2fd630228955f6d7e3794a98887db0d127a1fa0b4d5f4a042325b755cb4ce6
3
+ size 153260
videos/82984_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:e3aad88f160923fdd2a76cea40f34022e53ba062f30c350205458a9d64776982
3
+ size 153888
videos/82984_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:79ab75a577992280fbd911b28c04d23ffdd714027c8eb7acde0d54ac4745d07b
3
+ size 154500
videos/82984_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:fc9b5c67cf9cd70984243a025db9f8ac6e8684b1266ed2e05d74aa282a63fb6b
3
+ size 150990
videos/82984_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:7d37192d86390adbc212eb2163e172ddac7db5c1165fa903d20715f381d5c3b8
3
+ size 159302
videos/82984_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:a6cf2e3f948c8c20efed5918f9d00e6ac4f53eebef9eb740f7b400651de070bd
3
+ size 154168
videos/82984_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:c3d4464b1d3bbd878c37528758904b2159ac79ffea62dbe29909983646b05179
3
+ size 159344
videos/82984_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:c1635630869c3c8401d205218484bbb5ff6b5d01338029954151f12ff9f829fe
3
+ size 153563
videos/82984_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:b0de107c496e01fc80b8f35cb788d4e3150ff637e38a3505a66449b71e8f2c5d
3
+ size 151933
videos/82984_0/epoch_20/3_traj_1000003_envseed.mp4 ADDED
Binary file (82.5 kB). View file
 
videos/82984_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:524236ea22e6b37f6f41706b9e83ed32db4452c48e768bf8214af7cc5bfbd907
3
+ size 157297
videos/82984_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:04ebfa219c75b2bfaa5d13bfc0698ff6337be59b7c5b33129dbb93a4ed754137
3
+ size 153549
videos/82984_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:2746087fb2ee54c0c00e4e402c7a6b0530d0c999a5daa3d450cc901bef8ced7f
3
+ size 154116
videos/82984_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:26e64e317427f915541ce480ec856b0ea21211f25a069480d97e89e5c2913df7
3
+ size 162473
videos/82984_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:85edd8ab5cd417c618c676a04891685464f1cac7629534f71d951f7d26f2262a
3
+ size 159657
videos/82984_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:20262b480e3464970e93e2562b99310663e6ac6e0d542e20e20ada65f9869067
3
+ size 153836
videos/82984_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:0630dfff908f89d387c5a5258fee9b8332bc052be73728c789e5d8e14bfa460f
3
+ size 152062
videos/82984_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:152fe54d6d3590fc741bd6d115bdebfdd767296fcbafcd3b6b4beb79afa58503
3
+ size 153949
videos/82984_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:2a34c7e5700683a68978e09f9d985e2d04bc7a605611c0e33c6e0a6cb144662c
3
+ size 154723
videos/82984_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:7d60d98665bdd1f06c4fcb8eee4a16fc9bf7f6dc097dd33a9b0618c0035a9949
3
+ size 154524
videos/82984_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:c2018e32e2789e02c758f12cf9f6a4d9c6f4fd57b6ba0be3a5715145273cc773
3
+ size 153599
videos/82984_0/epoch_20/50_traj_1000050_envseed.mp4 ADDED
Binary file (85.1 kB). View file
 
videos/82984_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:f83528e0425df96b26947dce84fc634f085496e465e2fba942071a516bae0918
3
+ size 154832
videos/82984_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:e2df7fe4dbce08a438aa82d5c65b1c3a7525bee0134505b7c4f9174002370ae8
3
+ size 156391