Functions — github.com/bhpfelix/segment-anything-finetuner

CodeHub MCPcopy Create free account

hub / github.com/bhpfelix/segment-anything-finetuner / functions

Functions17 in github.com/bhpfelix/segment-anything-finetuner

⨍ Functions 17 ◇ Types & classes 2

↓ 1 callersFunctionall_gather

Run all_gather on arbitrary picklable data (not necessarily tensors) Args: data: any picklable object Returns: list[data]

↓ 1 callersFunctionget_world_size

↓ 1 callersFunctionmain

finetune.py:335

Method__getitem__

finetune.py:107

(self, data_root, split, image_size)

( self, model_type, checkpoint_path, freeze_image_encoder=Fals

finetune.py:145

finetune.py:104

Methodcollate_fn

finetune.py:137

Methodconfigure_optimizers

finetune.py:290

(self, imgs, bboxes, labels)

finetune.py:186

Methodtrain_dataloader

finetune.py:316

Methodtraining_step

(self, batch, batch_nb)

finetune.py:247

Methodval_dataloader

finetune.py:325

Methodvalidation_epoch_end

(self, outputs)

finetune.py:273

Methodvalidation_step

(self, batch, batch_nb)

finetune.py:267

Methodwarmup_step_lr

finetune.py:293

Methodwarmup_step_lr_builder

(warmup_steps, milestones, gamma)

finetune.py:292