↓ 1 callersMethodbuild_overlapping_cropsDecompose an image into a set of overlapping crops :return crop_arr: [n_crops, h, w, 3] The crops :return mask_arr: [n_crops, h, w] T
olmo/preprocessing/image_preprocessor.py:451
↓ 1 callersFunctionbuild_qualitative_table(
name,
split,
n,
preprocessor,
shuffle=True,
show_patches=False,
show_crops=False
scripts/dataset_visualize.py:29
↓ 1 callersFunctioncross_entropy_loss(
logits, labels, ignore_index: int = -100, reduction: str = "mean", compute_z_loss: bool = False, z_loss_
olmo/train/trainer.py:261
↓ 1 callersFunctiondispatch_ring_flash_attn(
q: torch.Tensor,
k: torch.Tensor,
v: torch.Tensor,
*,
group: dist.ProcessGroup,
stra
olmo/nn/flash_attention_api.py:116
↓ 1 callersMethoddownload(cls, n_procs=1, check_sha=False, n_val=2048, cache_only=False, sample=None)
olmo/data/pixmo_datasets.py:367