saliacoel commited on
Commit
a336c71
·
verified ·
1 Parent(s): f5d70cd

Upload Custom_Batch_Output.py

Browse files
Files changed (1) hide show
  1. Custom_Batch_Output.py +1 -1
Custom_Batch_Output.py CHANGED
@@ -4,7 +4,7 @@
4
  import torch
5
 
6
 
7
- class SpecialBatchSplit:
8
  """
9
  Input:
10
  - images (IMAGE batch, typically torch.Tensor [B, H, W, C])
 
4
  import torch
5
 
6
 
7
+ class Custom_Batch_Output:
8
  """
9
  Input:
10
  - images (IMAGE batch, typically torch.Tensor [B, H, W, C])