Slep commited on
Commit
014beea
1 Parent(s): 98e6bb9

Remove misleading comment.

Browse files
Files changed (1) hide show
  1. src/model.py +1 -1
src/model.py CHANGED
@@ -154,7 +154,7 @@ class ConditionalViT(nn.Module):
154
  def forward(self, imgs: torch.Tensor, c: torch.Tensor = None):
155
  """
156
  imgs : Batch of images
157
- c : category indices. 0 = "No given category".
158
  """
159
 
160
  x = self.conv1(imgs) # shape = [*, width, grid, grid]
 
154
  def forward(self, imgs: torch.Tensor, c: torch.Tensor = None):
155
  """
156
  imgs : Batch of images
157
+ c : category indices.
158
  """
159
 
160
  x = self.conv1(imgs) # shape = [*, width, grid, grid]