Functions — github.com/KaiyangZhou/pytorch-center-loss

CodeHub MCPcopy Create free account

hub / github.com/KaiyangZhou/pytorch-center-loss / functions

Functions24 in github.com/KaiyangZhou/pytorch-center-loss

⨍ Functions 24 ◇ Types & classes 6

↓ 3 callersMethodclose

↓ 3 callersMethodupdate

(self, val, n=1)

↓ 2 callersFunctionmkdir_if_missing

↓ 2 callersFunctionplot_features

Plot features on 2D plane. Args: features: (num_instances, num_features). labels: (num_instances).

↓ 1 callersFunctionmain

↓ 1 callersMethodreset

↓ 1 callersFunctiontest

(model, testloader, use_gpu, num_classes, epoch)

↓ 1 callersFunctiontrain

(model, criterion_xent, criterion_cent, optimizer_model, optimizer_centloss, trainloader,

transforms.py:8

Method__enter__

(self, num_classes=10, feat_dim=2, use_gpu=True)

center_loss.py:14

(self, fpath=None)

(self, num_classes)

(self, batch_size, use_gpu, num_workers)

(name, num_classes)

(name, batch_size, use_gpu, num_workers)

Args: x: feature matrix with shape (batch_size, feat_dim). labels: ground truth labels with shape (batch_size).

center_loss.py:25

Functionsave_checkpoint

(state, is_best, fpath='checkpoint.pth.tar')