Commit
·
727445c
1
Parent(s):
29ce656
set vertices and faces to zeros
Browse files- utils/common_viz.py +2 -0
utils/common_viz.py
CHANGED
|
@@ -103,6 +103,8 @@ def get_batch(
|
|
| 103 |
batch["caption_feat"] = caption_feat
|
| 104 |
batch['char_feat'] = torch.zeros_like(batch['char_feat'])
|
| 105 |
batch['char_raw']['char_raw_feat'] = torch.zeros_like(batch['char_raw']['char_raw_feat'])
|
|
|
|
|
|
|
| 106 |
|
| 107 |
return batch
|
| 108 |
|
|
|
|
| 103 |
batch["caption_feat"] = caption_feat
|
| 104 |
batch['char_feat'] = torch.zeros_like(batch['char_feat'])
|
| 105 |
batch['char_raw']['char_raw_feat'] = torch.zeros_like(batch['char_raw']['char_raw_feat'])
|
| 106 |
+
batch['char_raw']['char_vertices'] = torch.zeros_like(batch['char_raw']['char_vertices'])
|
| 107 |
+
batch['char_raw']['char_faces'] = torch.zeros_like(batch['char_raw']['char_faces'])
|
| 108 |
|
| 109 |
return batch
|
| 110 |
|