Functions — github.com/TorchSSL/TorchSSL

CodeHub MCPcopy Create free account

hub / github.com/TorchSSL/TorchSSL / functions

Functions367 in github.com/TorchSSL/TorchSSL

⨍ Functions 367◇ Types & classes 53

models/vat/vat_utils.py:14

models/freematch/freematch_utils.py:13

models/uda/uda_utils.py:15

models/pseudolabel/pseudolabel_utils.py:14

models/fullysupervised/fullysupervised_utils.py:13

models/softmatch/softmatch_utils.py:13

Method__getattr__

custom_writer.py:151

Method__getitem__

If strong augmentation is not used, return weak_augment_image, target else: return weak_augment_image, strong

datasets/dataset.py:57

Method__getitem__

datasets/ssl_dataset.py:90

(self, log_dir, enabled=True)

custom_writer.py:17

(self, params, lr=required, momentum=0, dampening=0, weight_decay=0, nesterov=False)

train_utils.py:70

(self, tb_dir, file_name, use_tensorboard=False)

train_utils.py:139

train_utils.py:172

(self, model, decay)

train_utils.py:351

freeze_bn and unfreeze_bn must appear in pairs

train_utils.py:389

Args data: x_data targets: y_data (if not exist, None) num_classes: number of label classes t

datasets/dataset.py:20

(self, root, transform, ulb, num_labels=-1)

datasets/ssl_dataset.py:64

(self, root_path, num_labels=-1, num_class=1000)

datasets/ssl_dataset.py:144

(self, sampler, num_replicas=None, rank=None)

datasets/DistributedProxySampler.py:26

datasets/augmentation/randaugment.py:165

models/freematch_entropy/freematch_utils.py:7

class Freematch contains setter of data_loader, optimizer, and model update methods. Args: net_builder: backbone network

models/freematch_entropy/freematch.py:24

class Fixmatch contains setter of data_loader, optimizer, and model update methods. Args: net_builder: backbone network c

models/fixmatch/fixmatch.py:23

models/fixmatch/fixmatch_utils.py:7

models/mixmatch/mixmatch_utils.py:8

class Mixmatch contains setter of data_loader, optimizer, and model update methods. Args: net_builder: backbone network c

models/mixmatch/mixmatch.py:20

class Flexmatch contains setter of data_loader, optimizer, and model update methods. Args: net_builder: backbone network

models/flexmatch/flexmatch.py:23

models/flexmatch/flexmatch_utils.py:10

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

models/nets/resnet50.py:24

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

models/nets/resnet50.py:79

(self, is_remix=False)

models/nets/resnet50.py:246

(self, num_features, alpha=0.1, eps=1e-05, momentum=0.001, affine=True, track_running_stats=True)

models/nets/wrn.py:17

(self, nb_layers, in_planes, out_planes, block, stride, drop_rate=0.0, activate_before_residual=False)

models/nets/wrn.py:55

(self, first_stride, num_classes, depth=28, widen_factor=2, drop_rate=0.0, is_remix=False)

models/nets/wrn.py:72

(self, first_stride=1, depth=28, widen_factor=2, bn_momentum=0.01, leaky_slope=0.0, dropRate=0.0,

models/nets/wrn.py:132

(self, num_features, alpha=0.1, eps=1e-05, momentum=0.001, affine=True, track_running_stats=True)

models/nets/wrn_var.py:17

(self, nb_layers, in_planes, out_planes, block, stride, drop_rate=0.0, activate_before_residual=False)

models/nets/wrn_var.py:55

(self, first_stride, num_classes, depth=28, widen_factor=2, drop_rate=0.0, is_remix=False)

models/nets/wrn_var.py:72

(self, first_stride=1, depth=28, widen_factor=2, bn_momentum=0.01, leaky_slope=0.0, dropRate=0.0,

models/nets/wrn_var.py:136

class Fixmatch contains setter of data_loader, optimizer, and model update methods. Args: net_builder: backbone network c

models/remixmatch/remixmatch.py:21

models/remixmatch/remixmatch_utils.py:7

models/meanteacher/meanteacher_utils.py:7

class Fixmatch contains setter of data_loader, optimizer, and model update methods. Args: net_builder: backbone network c

models/meanteacher/meanteacher.py:20

models/pimodel/pimodel_utils.py:7

class PiModel contains setter of data_loader, optimizer, and model update methods. Args: net_builder: backbone network cl

models/pimodel/pimodel.py:17

models/vat/vat_utils.py:8

class Vat contains setter of data_loader, optimizer, and model update methods. Args: net_builder: backbone network class

models/vat/vat.py:18

models/freematch/freematch_utils.py:7

class Freematch contains setter of data_loader, optimizer, and model update methods. Args: net_builder: backbone network

models/freematch/freematch.py:24

models/uda/uda_utils.py:9

class Uda contains setter of data_loader, optimizer, and model update methods. Args: net_builder: backbone network class

models/uda/uda.py:20

class PseudoLabel contains setter of data_loader, optimizer, and model update methods. Args: net_builder: backbone networ

models/pseudolabel/pseudolabel.py:18

models/pseudolabel/pseudolabel_utils.py:8

class FullySupervised contains setter of data_loader, optimizer, and model update methods. Args: net_builder: backbone ne

models/fullysupervised/fullysupervised.py:16

models/fullysupervised/fullysupervised_utils.py:7

class Freematch contains setter of data_loader, optimizer, and model update methods. Args: net_builder: backbone network

models/softmatch/softmatch.py:24

models/softmatch/softmatch_utils.py:7

datasets/DistributedProxySampler.py:30

datasets/dataset.py:112

Method__setstate__

train_utils.py:85

Function_lr_lambda

_lr_lambda returns a multiplicative factor given an interger parameter epochs. Decaying criteria: last_epoch

train_utils.py:234

Functionaccuracy

Computes the accuracy over the k top predictions for the specified values of k Args output: logits or probs (num of batch, num o

train_utils.py:263

(self, num_classes)

models/nets/resnet50.py:249

Functionconsistency_loss

models/remixmatch/remixmatch_utils.py:35

Functiondefault_loader

datasets/ssl_dataset.py:55

Functionepoch2iter

train_utils.py:255

(self, x: Tensor)

models/nets/resnet50.py:51

(self, x: Tensor)

models/nets/resnet50.py:105

models/nets/resnet50.py:241

models/nets/wrn.py:21

models/nets/wrn.py:67

(self, x, ood_test=False)

models/nets/wrn.py:111

models/nets/wrn_var.py:21

models/nets/wrn_var.py:67

(self, x, ood_test=False)

models/nets/wrn_var.py:114

Functionget_imagenet_schedule

(optimizer, num_training_steps, num_labels, batch_size)

train_utils.py:250

Methodget_value

models/freematch_entropy/freematch_utils.py:10

Methodget_value

models/fixmatch/fixmatch_utils.py:10

Methodget_value

models/mixmatch/mixmatch_utils.py:11

Methodget_value

models/flexmatch/flexmatch_utils.py:13

Methodget_value

models/remixmatch/remixmatch_utils.py:10

Methodget_value

models/meanteacher/meanteacher_utils.py:10

Methodget_value

models/pimodel/pimodel_utils.py:10

Methodget_value

models/vat/vat_utils.py:11

Methodget_value

models/freematch/freematch_utils.py:10

Methodget_value

models/uda/uda_utils.py:12

Methodget_value

models/pseudolabel/pseudolabel_utils.py:11

Methodget_value

models/fullysupervised/fullysupervised_utils.py:10

Methodget_value

models/softmatch/softmatch_utils.py:10

Methodinterleave

(self, xy, batch)

models/freematch_entropy/freematch.py:414

Methodinterleave

(self, xy, batch)

models/fixmatch/fixmatch.py:306

Methodinterleave

(self, xy, batch)

models/flexmatch/flexmatch.py:324

Methodinterleave

(self, xy, batch)

models/meanteacher/meanteacher.py:276

Methodinterleave

(self, xy, batch)

models/pimodel/pimodel.py:268

Methodinterleave

(self, xy, batch)

models/vat/vat.py:263

Methodinterleave

(self, xy, batch)

models/freematch/freematch.py:411

Methodinterleave

(self, xy, batch)

models/uda/uda.py:308

Methodinterleave

(self, xy, batch)

models/pseudolabel/pseudolabel.py:285

Methodinterleave

(self, xy, batch)

models/fullysupervised/fullysupervised.py:253

← previous next →201–300 of 367, ranked by callers