revised character position

utils/common_viz.py

@@ -108,8 +108,8 @@ def get_batch(
         device=device,
     )
 
-    batch['char_feat'] = character_position.unsqueeze(0).repeat(1, 300).unsqueeze(0)
-    batch['char_raw']['char_raw_feat'] = character_position.unsqueeze(0).repeat(1, 300).unsqueeze(0)
+    batch['char_feat'] = character_position.unsqueeze(-1).repeat(1, 1, 300)
+    batch['char_raw']['char_raw_feat'] = character_position.unsqueeze(-1).repeat(1, 1, 300)
     batch['char_raw']['char_vertices'] = torch.zeros_like(batch['char_raw']['char_vertices'])
     batch['char_raw']['char_faces'] = torch.zeros_like(batch['char_raw']['char_faces'])