brails.processors.foundation_classifier.npid.main module
- brails.processors.foundation_classifier.npid.main.accuracy(output, target, topk=(1,))
Computes the precision@k for the specified values of k
- brails.processors.foundation_classifier.npid.main.adjust_learning_rate(optimizer, epoch)
Sets the learning rate to the initial LR decayed by 10 every 100 epochs
- brails.processors.foundation_classifier.npid.main.main()
- brails.processors.foundation_classifier.npid.main.save_checkpoint(state, is_best, prefix, filename='checkpoint.pth.tar')
- brails.processors.foundation_classifier.npid.main.train(train_loader, model, lemniscate, criterion, optimizer, epoch)