Functions — github.com/TorchSSL/TorchSSL

hub / github.com/TorchSSL/TorchSSL / functions

Functions367 in github.com/TorchSSL/TorchSSL

⨍ Functions 367◇ Types & classes 53

↓ 1 callersMethodcal_time_p_and_p_model

(self,logits_x_ulb_w, time_p, p_model, label_hist)

models/freematch/freematch.py:317

↓ 1 callersMethodcalculate_mask

(self, probs)

models/softmatch/softmatch.py:253

↓ 1 callersFunctionconsistency_loss

(dataset, logits_s, logits_w,time_p,p_model, name='ce', use_hard_labels=True)

models/freematch_entropy/freematch_utils.py:54

↓ 1 callersFunctionconsistency_loss

(logits_s, logits_w, name='ce', T=1.0, p_cutoff=0.0, use_hard_labels=True)

models/fixmatch/fixmatch_utils.py:17

↓ 1 callersFunctionconsistency_loss

(logits_w, y)

models/mixmatch/mixmatch_utils.py:40

↓ 1 callersFunctionconsistency_loss

(logits_s, logits_w, class_acc, p_target, p_model, name='ce', T=1.0, p_cutoff=0.0, use_ha

models/flexmatch/flexmatch_utils.py:20

↓ 1 callersFunctionconsistency_loss

(logits_w1, logits_w2)

models/meanteacher/meanteacher_utils.py:17

↓ 1 callersFunctionconsistency_loss

(logits_w1, logits_w2)

models/pimodel/pimodel_utils.py:17

↓ 1 callersFunctionconsistency_loss

(dataset,logits_s, logits_w,time_p,p_model, name='ce', use_hard_labels=True)

models/freematch/freematch_utils.py:18

↓ 1 callersFunctionconsistency_loss

(logits_s, logits_w, class_acc, it, ds, name='ce', T=1.0, p_cutoff=0.0, use_flex=False)

models/uda/uda_utils.py:36

↓ 1 callersFunctionconsistency_loss

(logits_w, class_acc, it, ds, p_cutoff, use_flex=False)

models/pseudolabel/pseudolabel_utils.py:18

↓ 1 callersFunctionconsistency_loss

(logits_s, logits_w, mask, name='ce', T=0.5, use_hard_labels=True)

models/softmatch/softmatch_utils.py:18

↓ 1 callersMethoddistribution_alignment

(self, probs)

models/softmatch/softmatch.py:263

↓ 1 callersMethoddump_stats

(self)

custom_writer.py:43

↓ 1 callersFunctionentropy_loss

(mask, logits_s, logits_w, prob_model, label_hist)

models/freematch_entropy/freematch_utils.py:24

↓ 1 callersFunctionentropy_loss

(ul_y)

models/vat/vat_utils.py:17

↓ 1 callersFunctionexp_baseline

(label_amount)

scripts/config_generator.py:178

↓ 1 callersFunctionexp_flex_component

(label_amount)

scripts/config_generator.py:250

↓ 1 callersMethodforward

(self, x)

models/nets/wrn.py:42

↓ 1 callersMethodforward

(self, x)

models/nets/wrn_var.py:42

↓ 1 callersFunctionget_onehot

(num_classes, idx)

datasets/data_utils.py:147

↓ 1 callersFunctionget_sampler_by_name

get sampler in torch.utils.data.sampler by name

datasets/data_utils.py:69

↓ 1 callersFunctionget_static

(file_name)

scripts/average_log.py:8

↓ 1 callersFunctionget_transform

(mean, std, crop_size, train=True)

datasets/ssl_dataset.py:180

↓ 1 callersMethodinterleave_offsets

(self, batch, nu)

models/freematch_entropy/freematch.py:404

↓ 1 callersMethodinterleave_offsets

(self, batch, nu)

models/fixmatch/fixmatch.py:296

↓ 1 callersMethodinterleave_offsets

(self, batch, nu)

models/mixmatch/mixmatch.py:300

↓ 1 callersMethodinterleave_offsets

(self, batch, nu)

models/flexmatch/flexmatch.py:314

↓ 1 callersMethodinterleave_offsets

(self, batch, nu)

models/remixmatch/remixmatch.py:347

↓ 1 callersMethodinterleave_offsets

(self, batch, nu)

models/meanteacher/meanteacher.py:266

↓ 1 callersMethodinterleave_offsets

(self, batch, nu)

models/pimodel/pimodel.py:258

↓ 1 callersMethodinterleave_offsets

(self, batch, nu)

models/vat/vat.py:253

↓ 1 callersMethodinterleave_offsets

(self, batch, nu)

models/freematch/freematch.py:401

↓ 1 callersMethodinterleave_offsets

(self, batch, nu)

models/uda/uda.py:298

↓ 1 callersMethodinterleave_offsets

(self, batch, nu)

models/pseudolabel/pseudolabel.py:275

↓ 1 callersMethodinterleave_offsets

(self, batch, nu)

models/fullysupervised/fullysupervised.py:243

↓ 1 callersFunctionmain

For (Distributed)DataParallelism, main(args) spawn each process (main_worker) to each GPU.

freematch_entropy.py:22

↓ 1 callersFunctionmain

For (Distributed)DataParallelism, main(args) spawn each process (main_worker) to each GPU.

remixmatch.py:22

↓ 1 callersFunctionmain

For (Distributed)DataParallelism, main(args) spawn each process (main_worker) to each GPU.

fixmatch.py:22

↓ 1 callersFunctionmain

For (Distributed)DataParallelism, main(args) spawn each process (main_worker) to each GPU.

pseudolabel.py:22

↓ 1 callersFunctionmain

For (Distributed)DataParallelism, main(args) spawn each process (main_worker) to each GPU.

fullysupervised.py:22

↓ 1 callersFunctionmain

For (Distributed)DataParallelism, main(args) spawn each process (main_worker) to each GPU.

uda.py:22

↓ 1 callersFunctionmain

For (Distributed)DataParallelism, main(args) spawn each process (main_worker) to each GPU.

flexmatch.py:22

↓ 1 callersFunctionmain

For (Distributed)DataParallelism, main(args) spawn each process (main_worker) to each GPU.

pimodel.py:22

↓ 1 callersFunctionmain

For (Distributed)DataParallelism, main(args) spawn each process (main_worker) to each GPU.

vat.py:22

↓ 1 callersFunctionmain

For (Distributed)DataParallelism, main(args) spawn each process (main_worker) to each GPU.

softmatch.py:22

↓ 1 callersFunctionmain

For (Distributed)DataParallelism, main(args) spawn each process (main_worker) to each GPU.

mixmatch.py:22

↓ 1 callersFunctionmain

For (Distributed)DataParallelism, main(args) spawn each process (main_worker) to each GPU.

meanteacher.py:22

↓ 1 callersFunctionmain

For (Distributed)DataParallelism, main(args) spawn each process (main_worker) to each GPU.

freematch.py:22

↓ 1 callersFunctionmain_worker

main_worker is conducted on each GPU.

freematch_entropy.py:66

↓ 1 callersFunctionmain_worker

main_worker is conducted on each GPU.

remixmatch.py:69

↓ 1 callersFunctionmain_worker

main_worker is conducted on each GPU.

fixmatch.py:69

↓ 1 callersFunctionmain_worker

main_worker is conducted on each GPU.

pseudolabel.py:69

↓ 1 callersFunctionmain_worker

main_worker is conducted on each GPU.

fullysupervised.py:69

↓ 1 callersFunctionmain_worker

main_worker is conducted on each GPU.

uda.py:69

↓ 1 callersFunctionmain_worker

main_worker is conducted on each GPU.

flexmatch.py:69

↓ 1 callersFunctionmain_worker

main_worker is conducted on each GPU.

pimodel.py:69

↓ 1 callersFunctionmain_worker

main_worker is conducted on each GPU.

vat.py:69

↓ 1 callersFunctionmain_worker

main_worker is conducted on each GPU.

softmatch.py:66

↓ 1 callersFunctionmain_worker

main_worker is conducted on each GPU.

mixmatch.py:69

↓ 1 callersFunctionmain_worker

main_worker is conducted on each GPU.

meanteacher.py:69

↓ 1 callersFunctionmain_worker

main_worker is conducted on each GPU.

freematch.py:66

↓ 1 callersMethodmake_dataset

( self, directory, class_to_idx, extensions=None,

datasets/ssl_dataset.py:100

↓ 1 callersMethodmatplotlib_plot

Plot stats using Matplotlib and save images.

custom_writer.py:90

↓ 1 callersFunctionmixup_one_target

Returns mixed inputs, mixed targets, and lambda

models/mixmatch/mixmatch_utils.py:23

↓ 1 callersFunctionmixup_one_target

Returns mixed inputs, mixed targets, and lambda

models/remixmatch/remixmatch_utils.py:20

↓ 1 callersFunctionone_hot

(targets, nClass, gpu)

models/mixmatch/mixmatch_utils.py:18

↓ 1 callersFunctionone_hot

(targets, nClass, gpu)

models/remixmatch/remixmatch_utils.py:16

↓ 1 callersMethodplot_stats

(self)

custom_writer.py:87

↓ 1 callersMethodreset

(self)

train_utils.py:175

↓ 1 callersMethodset_epoch

Execute this function to update the step attribute and compute the cost time of one epoch in seconds. Recommend to run this function

custom_writer.py:51

↓ 1 callersFunctionsplit_ssl_data

data & target is splitted into labeled and unlabeld data. Args index: If np.array of index is given, select the data[index], tar

datasets/data_utils.py:14

↓ 1 callersMethodto_numpy

(self, a)

custom_writer.py:135

↓ 1 callersMethodupdate_prob_t

(self, lb_probs, ulb_probs)

models/softmatch/softmatch.py:239

↓ 1 callersFunctionvat_loss

(model, ul_x, ul_y, xi=1e-6, eps=6, num_iters=1)

models/vat/vat_utils.py:37

FunctionAutoContrast

(img, _)

datasets/augmentation/randaugment.py:15

FunctionBrightness

(img, v)

datasets/augmentation/randaugment.py:19

FunctionColor

(img, v)

datasets/augmentation/randaugment.py:24

FunctionContrast

(img, v)

datasets/augmentation/randaugment.py:29

FunctionEqualize

(img, _)

datasets/augmentation/randaugment.py:34

FunctionIdentity

(img, v)

datasets/augmentation/randaugment.py:42

FunctionInvert

(img, _)

datasets/augmentation/randaugment.py:38

FunctionPosterize

(img, v)

datasets/augmentation/randaugment.py:46

FunctionRotate

(img, v)

datasets/augmentation/randaugment.py:52

FunctionSharpness

(img, v)

datasets/augmentation/randaugment.py:60

FunctionShearX

(img, v)

datasets/augmentation/randaugment.py:65

FunctionShearY

(img, v)

datasets/augmentation/randaugment.py:72

FunctionSolarize

(img, v)

datasets/augmentation/randaugment.py:109

FunctionTranslateX

(img, v)

datasets/augmentation/randaugment.py:79

FunctionTranslateXabs

(img, v)

datasets/augmentation/randaugment.py:87

FunctionTranslateY

(img, v)

datasets/augmentation/randaugment.py:94

FunctionTranslateYabs

(img, v)

datasets/augmentation/randaugment.py:102

Method__call__

(self, img)

datasets/augmentation/randaugment.py:171

Method__call__

(self, iter)

models/freematch_entropy/freematch_utils.py:13

Method__call__

(self, iter)

models/fixmatch/fixmatch_utils.py:13

Method__call__

(self, iter)

models/mixmatch/mixmatch_utils.py:14

Method__call__

(self, iter)

models/flexmatch/flexmatch_utils.py:16

Method__call__

(self, iter)

models/remixmatch/remixmatch_utils.py:13

Method__call__

(self, iter)

models/meanteacher/meanteacher_utils.py:13

Method__call__

(self, iter)

models/pimodel/pimodel_utils.py:13

← previous next →101–200 of 367, ranked by callers