Functions — github.com/andreasveit/densenet-pytorch

CodeHub MCPcopy Create free account

hub / github.com/andreasveit/densenet-pytorch / functions

Functions20 in github.com/andreasveit/densenet-pytorch

⨍ Functions 20 ◇ Types & classes 6

↓ 6 callersMethodupdate

(self, val, n=1)

↓ 4 callersMethod__init__

(self, depth, num_classes, growth_rate=12, reduction=0.5, bottleneck=True, dropRate=0.0)

↓ 2 callersFunctionaccuracy

Computes the precision@k for the specified values of k

↓ 1 callersMethod_make_layer

(self, block, in_planes, growth_rate, nb_layers, dropRate)

↓ 1 callersFunctionadjust_learning_rate

Sets the learning rate to the initial LR decayed by 10 after 150 and 225 epochs

↓ 1 callersFunctionmain

↓ 1 callersMethodreset

↓ 1 callersFunctionsave_checkpoint

Saves checkpoint to disk

↓ 1 callersFunctiontrain

Train for one epoch on the training set

↓ 1 callersFunctionvalidate

Perform validation on the validation set

(self, in_planes, out_planes, dropRate=0.0)

(self, in_planes, out_planes, dropRate=0.0)

(self, in_planes, out_planes, dropRate=0.0)

(self, nb_layers, in_planes, growth_rate, block, dropRate=0.0)

densenet.py:111