ray0rf1re
/

6net

@@ -1,7 +1,12 @@
 ---
 language: en
-tags: [robotics, 6-axis-arm, visual-policy, pytorch, imitation-learning]
-license: apache-2.0
 ---
 # 6Net — 6-Axis Visual Robot Policy (~115M)
@@ -37,4 +42,4 @@ img = tf(Image.open("cam.jpg")).unsqueeze(0)   # (1,3,224,224)
 jts = torch.zeros(1, 6)                         # current joint angles (rad)
 with torch.no_grad():
     action = model(img, jts)                     # (1,6) predicted targets
-```

 ---
 language: en
+tags:
+- robotics
+- 6-axis-arm
+- visual-policy
+- pytorch
+- imitation-learning
+license: mit
 ---
 # 6Net — 6-Axis Visual Robot Policy (~115M)
 jts = torch.zeros(1, 6)                         # current joint angles (rad)
 with torch.no_grad():
     action = model(img, jts)                     # (1,6) predicted targets
+```