↓ 1 callersFunctionprofile_fvcore(
model,
image_input_size=(3, 224, 224),
text_input_size=(77,),
batch_size=1,
biomedclip_finetuning/open_clip/src/open_clip_train/profiler.py:23
↓ 1 callersFunctionprofile_fvcore_image(
model,
image_input_size=(3, 224, 224),
batch_size=1,
detailed=False,
biomedclip_finetuning/open_clip/src/open_clip_train/profiler.py:63
↓ 1 callersFunctionprofile_fvcore_text(
model,
text_input_size=(77,),
batch_size=1,
detailed=False,
force_cp
biomedclip_finetuning/open_clip/src/open_clip_train/profiler.py:44
↓ 1 callersFunctionrun_training(dataset_name_or_id: Union[str, int],
configuration: str, fold: Union[int, str],
weak_segmentation/nnunetv2/run/run_training.py:143
↓ 1 callersFunctiontext_heatmap_iba(text_t, image_t, model, layer_idx, beta, var, lr=1, train_steps=10, progbar=True)
saliency_maps/scripts/methods.py:45
↓ 1 callersFunctiontrain_one_epoch(model, data, loss, epoch, optimizer, scaler, scheduler, dist_model, args, tb_writer=None)
biomedclip_finetuning/open_clip/src/open_clip_train/train.py:64