Skip to content

Commit

Permalink
move vit functionality into derandomised smoothing toolset
Browse files Browse the repository at this point in the history
Signed-off-by: GiulioZizzo <giulio.zizzo@yahoo.co.uk>
  • Loading branch information
GiulioZizzo committed Jun 27, 2023
1 parent 284f590 commit 3dc8283
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -93,7 +93,7 @@ def __init__(
if original_shape is not None and output_shape is not None:
self.upsample = UpSampler(input_size=original_shape[1], final_size=output_shape[1])

def ablate(self, x: torch.Tensor, column_pos: int) -> torch.Tensor:
def ablate(self, x: torch.Tensor, column_pos: Optional[int] = None) -> torch.Tensor:
"""
Ablates the input colum wise
Expand Down

0 comments on commit 3dc8283

Please sign in to comment.