Functions — github.com/Nota-NetsPresso/netspresso-trainer

Method__init__

(self, dataset, num_replicas=None, rank=None, shuffle=False, seed=0)

src/netspresso_trainer/dataloaders/utils/sampler.py:77

Method__init__

( self, size: List, affine_scale: List, mixup_scale: List, degrees: fl

src/netspresso_trainer/dataloaders/augmentation/custom/mosaic.py:163

Method__init__

( self, num_classes: int, cutmix: Optional[List], mixup: Optional[List],

src/netspresso_trainer/dataloaders/augmentation/custom/mixing.py:30

Method__init__

( self, num_classes: int, alpha: float, p: float, inplace: bool, )

src/netspresso_trainer/dataloaders/augmentation/custom/mixing.py:81

Method__init__

( self, num_classes: int, alpha: float, p: float, inplace: bool, )

src/netspresso_trainer/dataloaders/augmentation/custom/mixing.py:164

Method__init__

(self, transforms, additional_targets: Dict = None)

src/netspresso_trainer/dataloaders/augmentation/custom/image_proc.py:44

Method__init__

( self, size: Union[int, List], )

src/netspresso_trainer/dataloaders/augmentation/custom/image_proc.py:89

Method__init__

(self)

src/netspresso_trainer/dataloaders/augmentation/custom/image_proc.py:106

Method__init__

( self, size: Union[int, List], interpolation: str, max_size: Optional[int],

src/netspresso_trainer/dataloaders/augmentation/custom/image_proc.py:119

Method__init__

( self, p: float, )

src/netspresso_trainer/dataloaders/augmentation/custom/image_proc.py:162

Method__init__

( self, p: float, )

src/netspresso_trainer/dataloaders/augmentation/custom/image_proc.py:188

Method__init__

( self, brightness: Union[float, List], contrast: Union[float, List], saturati

src/netspresso_trainer/dataloaders/augmentation/custom/image_proc.py:267

Method__init__

( self, size: Union[int, List], fill: int, )

src/netspresso_trainer/dataloaders/augmentation/custom/image_proc.py:307

Method__init__

( self, size: Union[int, List], scale: Union[float, List], ratio: Union[float,

src/netspresso_trainer/dataloaders/augmentation/custom/image_proc.py:358

Method__init__

( self, p: float, scale: List, ratio: List, value: Optional[int],

src/netspresso_trainer/dataloaders/augmentation/custom/image_proc.py:418

Method__init__

( self, min_scale: float, max_scale: float, min_aspect_ratio: float, m

src/netspresso_trainer/dataloaders/augmentation/custom/image_proc.py:476

Method__init__

( self, fill: int, side_range: List[float], p: float, )

src/netspresso_trainer/dataloaders/augmentation/custom/image_proc.py:558

Method__init__

( self, num_magnitude_bins: int, interpolation: str, fill: Optional[Union[List

src/netspresso_trainer/dataloaders/augmentation/custom/image_proc.py:612

Method__init__

( self, policy: str, interpolation: str, fill: Optional[List[float]], )

src/netspresso_trainer/dataloaders/augmentation/custom/image_proc.py:682

Method__init__

( self, h_mag: int, s_mag: int, v_mag: int, )

src/netspresso_trainer/dataloaders/augmentation/custom/image_proc.py:726

Method__init__

( self, base_size: List, stride: int, random_range: int, interpolation

src/netspresso_trainer/dataloaders/augmentation/custom/image_proc.py:759

Method__init__

( self, base_size: List[int], random_range: Union[int, float], interpolation:

src/netspresso_trainer/dataloaders/augmentation/custom/image_proc.py:799

Method__init__

( self, scale: List, scale_prob: float, translate: float, translate_pr

src/netspresso_trainer/dataloaders/augmentation/custom/image_proc.py:837

Method__init__

(self, mean, std)

src/netspresso_trainer/dataloaders/augmentation/custom/image_proc.py:980

Method__init__

(self, pixel_range)

src/netspresso_trainer/dataloaders/augmentation/custom/image_proc.py:997

Method__init__

(self, optimizer: torch.optim.Optimizer, param_group_field: str,

src/netspresso_trainer/schedulers/base.py:41

Method__init__

( self, optimizer, scheduler_conf, training_epochs, )

src/netspresso_trainer/schedulers/cosine_lr.py:36

Method__init__

( self, optimizer, scheduler_conf, training_epochs, )

src/netspresso_trainer/schedulers/poly_lr.py:34

Method__init__

( self, optimizer, scheduler_conf, training_epochs, )

src/netspresso_trainer/schedulers/step_lr.py:53

Method__init__

( self, optimizer, scheduler_conf, training_epochs, )

src/netspresso_trainer/schedulers/cosine_warm_restart.py:60

Method__init__

(self, o)

src/netspresso_trainer/schedulers/cosine_warm_restart.py:178

Method__init__

( self, optimizer, scheduler_conf, training_epochs, )

src/netspresso_trainer/schedulers/multi_step_lr.py:42

Method__init__

( self, params, optimizer_conf, )

src/netspresso_trainer/optimizers/custom.py:21

Method__init__

( self, params, optimizer_conf, )

src/netspresso_trainer/optimizers/custom.py:34

Method__init__

( self, params, optimizer_conf, )

src/netspresso_trainer/optimizers/custom.py:47

Method__init__

( self, params, optimizer_conf, )

src/netspresso_trainer/optimizers/custom.py:60

Method__init__

( self, params, optimizer_conf, )

src/netspresso_trainer/optimizers/custom.py:73

Method__init__

( self, params, optimizer_conf, )

src/netspresso_trainer/optimizers/custom.py:86

Method__init__

(self, conf_model, backbone, neck, head, freeze_backbone=False)

src/netspresso_trainer/models/base.py:89

Method__init__

(self, conf_model, backbone, neck, head, freeze_backbone=False)

src/netspresso_trainer/models/base.py:99

Method__init__

(self, conf_model, backbone, neck, head, freeze_backbone=False)

src/netspresso_trainer/models/base.py:111

Method__init__

(self, conf_model, backbone, neck, head, freeze_backbone=False)

src/netspresso_trainer/models/base.py:123

Method__init__

(self, model_conf)

src/netspresso_trainer/models/base.py:138

Method__init__

(self, model_conf)

src/netspresso_trainer/models/base.py:181

Method__init__

( self, intermediate_features_dim: List[int], params: DictConfig, )

src/netspresso_trainer/models/necks/experimental/yolov9fpn.py:32

Method__init__

( self, intermediate_features_dim: List[int], params: DictConfig, )

src/netspresso_trainer/models/necks/experimental/fpn.py:174

Method__init__

( self, intermediate_features_dim: List[int], params: DictConfig, )

src/netspresso_trainer/models/necks/experimental/yolopafpn.py:35

Method__init__

(self, d_model, nhead, dim_feedforward=2048,

src/netspresso_trainer/models/necks/experimental/rtdetr_hybrid_encoder.py:36

Method__init__

(self, encoder_layer, num_layers, norm=None)

src/netspresso_trainer/models/necks/experimental/rtdetr_hybrid_encoder.py:84

Method__init__

( self, task: str, params: Optional[DictConfig] = None, stage_params: Optional

src/netspresso_trainer/models/backbones/core/resnet.py:44

Method__init__

(self, image_channels, patch_size, hidden_size, hidden_dropout_prob, use_cls_token=True, vocab_size=1000)

src/netspresso_trainer/models/backbones/experimental/vit.py:45

Method__init__

(self, num_blocks, hidden_size, num_attention_heads, attention_dropout_prob, intermediate_size, hidden_dropout

src/netspresso_trainer/models/backbones/experimental/vit.py:107

Method__init__

( self, task: str, params: Optional[DictConfig] = None, stage_params: Optional

src/netspresso_trainer/models/backbones/experimental/vit.py:115

Method__init__

( self, task: str, params: Optional[DictConfig] = None, stage_params: Optional

src/netspresso_trainer/models/backbones/experimental/darknet.py:37

Method__init__

( self, task: str, params: Optional[DictConfig] = None, stage_params: Optional

src/netspresso_trainer/models/backbones/experimental/gelan.py:33

Method__init__

( self, task: str, params: Optional[DictConfig] = None, stage_params: Optional

src/netspresso_trainer/models/backbones/experimental/mobilenetv3.py:39

Method__init__

(self, in_channels, out_channels)

src/netspresso_trainer/models/backbones/experimental/efficientformer.py:50

Method__init__

(self, in_channels, out_channels, patch_size, stride, padding)

src/netspresso_trainer/models/backbones/experimental/efficientformer.py:65

Method__init__

(self, hidden_size, intermediate_size=None, hidden_dropout_prob=0., hidden_activation_type='gelu')

src/netspresso_trainer/models/backbones/experimental/efficientformer.py:78

Method__init__

(self, hidden_size, pool_size=3, intermediate_ratio=4., hidden_dropout_prob=0., hidden_activa

src/netspresso_trainer/models/backbones/experimental/efficientformer.py:177

Method__init__

(self, use_intermediate_features, num_blocks, hidden_sizes, num_attention_heads, attention_hi

src/netspresso_trainer/models/backbones/experimental/efficientformer.py:217

Method__init__

( self, task: str, params: Optional[DictConfig] = None, stage_params: Optional

src/netspresso_trainer/models/backbones/experimental/efficientformer.py:351

Method__init__

( self, task: str, params: Optional[DictConfig] = None, stage_params: Optional

src/netspresso_trainer/models/backbones/experimental/mobilenetv4.py:52

Method__init__

( self, task: str, params: Optional[DictConfig] = None, stage_params: Optional

src/netspresso_trainer/models/backbones/experimental/shufflenetv2.py:31

Method__init__

(self, image_channels, hidden_size)

src/netspresso_trainer/models/backbones/experimental/mobilevit.py:42

Method__init__

(self, hidden_size, num_attention_heads, attention_dropout_prob, intermediate_size, hidden_dropout_prob, layer

src/netspresso_trainer/models/backbones/experimental/mobilevit.py:64

Method__init__

(self, num_transformer_blocks, in_channels, hidden_size, local_kernel_size, patch_h, patch_w,

src/netspresso_trainer/models/backbones/experimental/mobilevit.py:76

Method__init__

( self, params: Optional[DictConfig] = None, stage_params: Optional[List] = None,

src/netspresso_trainer/models/backbones/experimental/mobilevit.py:276

Method__init__

(self, patch_size, stride, in_channels, hidden_size, layer_norm_eps)

src/netspresso_trainer/models/backbones/experimental/mixtransformer.py:39

Method__init__

(self, intermediate_size=768)

src/netspresso_trainer/models/backbones/experimental/mixtransformer.py:62

Method__init__

(self, in_features, intermediate_size, hidden_dropout_prob, hidden_activation_type)

src/netspresso_trainer/models/backbones/experimental/mixtransformer.py:77

Method__init__

(self, num_blocks, hidden_size, num_attention_heads, attention_dropout_prob, sr_ratio,

src/netspresso_trainer/models/backbones/experimental/mixtransformer.py:127

Method__init__

( self, task: str, params: Optional[DictConfig] = None, stage_params: Optional

src/netspresso_trainer/models/backbones/experimental/mixtransformer.py:156

Method__init__

(self, in_planes, kernel_sizes, stride=1, dilate=1)

src/netspresso_trainer/models/backbones/experimental/mixnet.py:71

Method__init__

(self, in_planes, out_planes, expand_ratio, num_exp_groups, kernel_sizes, num_poi_groups, str

src/netspresso_trainer/models/backbones/experimental/mixnet.py:100

Method__init__

( self, task: str, params: Optional[DictConfig] = None, stage_params: Optional

src/netspresso_trainer/models/backbones/experimental/mixnet.py:170

Method__init__

(self, drop_prob: float = 0., scale_by_keep: bool = True)

src/netspresso_trainer/models/op/depth.py:44

Method__init__

( self, in_channels: int, out_channels: int, kernel_size: Union[int, Tuple[int

src/netspresso_trainer/models/op/custom.py:72

Method__init__

( self, in_channels: int, out_channels: int, kernel_size: Union[int, Tuple[int

src/netspresso_trainer/models/op/custom.py:155

Method__init__

(self, in_channels: int, out_channels: int, kernel_size: Un

src/netspresso_trainer/models/op/custom.py:193

Method__init__

(self, method: str = "max", kernel_size: int = 2, stride: O

src/netspresso_trainer/models/op/custom.py:287

Method__init__

( self, inplanes: int, planes: int, stride: int = 1, downsample: Optio

src/netspresso_trainer/models/op/custom.py:304

Method__init__

( self, inplanes: int, planes: int, stride: int = 1, downsample: Optio

src/netspresso_trainer/models/op/custom.py:360

Method__init__

(self, in_channels: int, out_channels: int, shortcut: bool

src/netspresso_trainer/models/op/custom.py:416

Method__init__

( self, in_channels: int, hidden_channels: int, out_channels: int, ker

src/netspresso_trainer/models/op/custom.py:450

Method__init__

( self, in_channels: int, hidden_channels: int, out_channels: int, ext

src/netspresso_trainer/models/op/custom.py:525

Method__init__

( self, d_model: int, max_len: Optional[int] = 5000, channels_last: Optional[b

src/netspresso_trainer/models/op/custom.py:640

Method__init__

( self, pool_type: Optional[str] = "mean", keep_dim: Optional[bool] = False, *

src/netspresso_trainer/models/op/custom.py:739

Method__init__

(self, in_channels, out_channels, ksize=1, stride=1, act_type="silu")

src/netspresso_trainer/models/op/custom.py:803

Method__init__

Args: in_channels (int): input channels. out_channels (int): output channels. n (int): number of Bottlene

src/netspresso_trainer/models/op/custom.py:845

Method__init__

(self, in_channels: int, out_channels: int, num_blocks: int

src/netspresso_trainer/models/op/custom.py:919

Method__init__

( self, in_channels, out_channels, kernel_sizes=(5, 9, 13), act_type="silu" )

src/netspresso_trainer/models/op/custom.py:955

Method__init__

( self, in_channels: int, out_channels: int, hidden_channels: int, ker

src/netspresso_trainer/models/op/custom.py:980

Method__init__

( self, in_channels, out_channels, shortcut=True, expansion=0.5,

src/netspresso_trainer/models/op/custom.py:1035

Method__init__

(self, dim: int, init_values: float = 1e-5, inplace: bool = False)

src/netspresso_trainer/models/op/custom.py:1067

Method__init__

Args: x (Union[Tensor, Proxy]): Input tensor Returns: Union[Tensor, Proxy]: Output tensor after pooling and

src/netspresso_trainer/models/op/custom.py:1144

Method__init__

( self, in_channels: int, out_channels: int, act_type: Optional[str]= 'silu',

src/netspresso_trainer/models/op/custom.py:1193

Method__init__

( self, in_channels: int, out_channels: int, kernel_sizes: Tuple[int] = (5, 9,

src/netspresso_trainer/models/op/custom.py:1221

Method__init__

( self, in_channels: int, out_channels: int, hidden_channels: Optional[int] =

src/netspresso_trainer/models/op/custom.py:1259

Method__init__

(self, reg_max: int=16)

src/netspresso_trainer/models/op/custom.py:1284

Functions1,065 in github.com/Nota-NetsPresso/netspresso-trainer