MCPcopy Create free account

hub / github.com/corenel/pytorch-adda / functions

Functions26 in github.com/corenel/pytorch-adda

↓ 11 callersFunctionmake_variable
Convert Tensor to Variable.
utils.py:14
↓ 4 callersFunctionget_data_loader
Get data loader by name.
utils.py:58
↓ 4 callersFunctioninit_model
Init models with cuda and weights.
utils.py:66
↓ 4 callersFunctionsave_model
Save trained model.
utils.py:85
↓ 2 callersFunctioneval_src
Evaluate classifier for source domain.
core/pretrain.py:74
↓ 2 callersFunctioneval_tgt
Evaluation for target encoder by source classifier on target dataset.
core/test.py:9
↓ 1 callersMethod__init__
Init LeNet encoder.
models/lenet.py:10
↓ 1 callersMethod_check_exists
Check if dataset is download and in right place.
datasets/usps.py:83
↓ 1 callersMethoddownload
Download dataset.
datasets/usps.py:87
↓ 1 callersFunctionget_mnist
Get MNIST dataset loader.
datasets/mnist.py:10
↓ 1 callersFunctionget_usps
Get USPS dataset loader.
datasets/usps.py:117
↓ 1 callersFunctioninit_random_seed
Init random seed.
utils.py:44
↓ 1 callersMethodload_samples
Load sample images from dataset.
datasets/usps.py:100
↓ 1 callersFunctiontrain_src
Train classifier for source domain.
core/pretrain.py:10
↓ 1 callersFunctiontrain_tgt
Train encoder for target domain.
core/adapt.py:13
Method__getitem__
Get images and target for data loader. Args: index (int): Index Returns: tuple: (image, target) where target
datasets/usps.py:64
Method__init__
Init USPS dataset.
datasets/usps.py:36
Method__init__
Init LeNet encoder.
models/lenet.py:43
Method__init__
Init discriminator.
models/discriminator.py:9
Method__len__
Return size of dataset.
datasets/usps.py:79
Functiondenormalize
Invert normalization, and then convert array into image.
utils.py:28
Methodforward
Forward the LeNet.
models/lenet.py:33
Methodforward
Forward the LeNet classifier.
models/lenet.py:48
Methodforward
Forward the discriminator.
models/discriminator.py:24
Functioninit_weights
Init weights for layers w.r.t. the original paper.
utils.py:34
Functionmake_cuda
Use CUDA if it's available.
utils.py:21