Commit
·
cbaa3f6
1
Parent(s):
1c8c09a
set char_raw to zero
Browse files- utils/common_viz.py +1 -0
utils/common_viz.py
CHANGED
|
@@ -102,6 +102,7 @@ def get_batch(
|
|
| 102 |
batch["caption_raw"] = [prompt]
|
| 103 |
batch["caption_feat"] = caption_feat
|
| 104 |
batch['char_feat'] = torch.zeros_like(batch['char_feat'])
|
|
|
|
| 105 |
|
| 106 |
return batch
|
| 107 |
|
|
|
|
| 102 |
batch["caption_raw"] = [prompt]
|
| 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 |
|