↓ 2 callersFunctioncompute_linear_tflops_per_batch_per_gpu(
batch_size,
seq_len,
in_features,
out_features,
count_backward=False,
use_grad_check
vision/m4/training/utils.py:656
↓ 2 callersMethodget_bar_chart_from_aggregate_metrics(
self, metrics_values, metrics_names, x_label_categories, x_label, y_label, bar_size=30
)
vision/m4/sourcing/data_collection/visualization/pair_stat_dashboard.py:352
↓ 2 callersFunctionget_dataloader_from_config(
tokenizer,
image_transforms,
image_seq_len,
seed,
config,
is_train,
rank=None,
vision/m4/training/dataset.py:507
↓ 2 callersFunctionget_dataloaders(
config: Parameters,
rank: int,
world_size: int,
tokenizer,
train_image_transforms,
v
vision/m4/training/dataset.py:156