Functions — github.com/LaRiffle/ariann

CodeHub MCPcopy Create free account

hub / github.com/LaRiffle/ariann / functions

Functions29 in github.com/LaRiffle/ariann

⨍ Functions 29 ◇ Types & classes 7

↓ 6 callersMethod__init__

(self, dataset, out_features=10)

↓ 4 callersFunctionencode

Depending on the setting, acts on a tensor - Do nothing OR - Transform to fixed precision OR - Secret

↓ 3 callersFunctiontest

(args, model, private_test_loader)

procedure.py:98

↓ 2 callersFunctionget_data_loaders

(args, kwargs, private=True)

↓ 2 callersFunctionget_model

(model_name, dataset, out_features)

↓ 2 callersFunctionget_number_classes

↓ 2 callersFunctionkill_processes

(worker_processes)

↓ 2 callersFunctionload_state_dict

(model, model_name, dataset)

↓ 2 callersFunctionone_hot_of

Transform to one hot tensor Example: [0, 3, 9] => [[1., 0., 0., 0., 0., 0., 0., 0., 0., 0.], [0., 0., 0., 1

↓ 2 callersFunctionrun

↓ 1 callersFunctionbuild_prepocessing

(model, dataset, batch_size, workers, args)

preprocess.py:164

↓ 1 callersFunctionforward

(optimizer, model, data, target)

procedure.py:44

↓ 1 callersFunctiontrain

(args, model, private_train_loader, optimizer, epoch)

procedure.py:30

(self, dataset, out_features)

(self, dataset, out_features)

(self, out_features=10)

(self, out_features=10)

Functionalexnet

(dataset, out_features)

Functionprofile

A gentle profiler

procedure.py:10

Functionresnet18

(dataset, out_features)

(dataset, out_features)

Functionwrapper

(args_, *args, **kwargs)

procedure.py:13