Functions — github.com/alanjeffares/discreteVAE

CodeHub MCPcopy Create free account

hub / github.com/alanjeffares/discreteVAE / functions

Functions9 in github.com/alanjeffares/discreteVAE

⨍ Functions 9 ◇ Types & classes 1

↓ 4 callersMethoddecode

↓ 3 callersMethodencode

↓ 3 callersMethodsample

↓ 2 callersFunctionget_dataloaders

(cuda, batch_size)

↓ 1 callersFunctionmain

↓ 1 callersFunctionplot_elbo

(train_elbo_ls, test_elbo_ls)

↓ 1 callersFunctiontest

(model, device, test_loader)

↓ 1 callersFunctiontrain

(epoch, model, device, train_loader, optimizer)

(self, nlat=4, ncat=8)