Make mask tensor containing indices of padded part. See description of make_non_pad_mask. Args: lengths (torch.Tensor): Batch of len

Make dataset and collator for supervised fine-tuning.

x: b * c * h * w Assuming x contains num_split**2 sub-squares concatenated along batch dimension, merge the sub-squares back to the original

Parse information from a line in a requirements text file.

Parse the package dependencies listed in a requirements file but strips specific versioning information. Args: fname (str): path to r

Calculates relaxed correctness. The correctness tolerates certain error ratio defined by max_relative_change. See https://arxiv.org/pdf/2203.

Repeat module N times. :param int N: repeat time :param function fn: function to generate module :return: repeated modules :rtype: Mu