Skip to content

Commit

Permalink
Merge remote-tracking branch 'origin/546-masking-time-step-segmentati…
Browse files Browse the repository at this point in the history
…on' into 546-masking-time-step-segmentation
  • Loading branch information
cwmeijer committed Jan 24, 2024
2 parents 31fc6fb + e1be7ff commit 728139a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion dianna/utils/maskers.py
Original file line number Diff line number Diff line change
Expand Up @@ -161,7 +161,7 @@ def _mask_bottom_ratio(float_mask: np.ndarray, p_keep: float) -> np.ndarray:
return flat_mask.reshape(float_mask.shape)


def _generate_interpolated_float_masks(input_size: int, p_keep: float,
def _generate_interpolated_float_masks_for_image(input_size: int, p_keep: float,
number_of_masks: int,
number_of_features: int):
"""Generates a set of random masks to mask the input data.
Expand Down

0 comments on commit 728139a

Please sign in to comment.