code is found on the following github link

  
      github.com
  
  
    anubhav4sachan/fingerprint-recognition/blob/master/training.py 24
from __future__ import print_function
import torch
import torch.nn as nn
import torch.nn.functional as F
import torch.optim as optim
import torchvision
import matplotlib.pyplot as plt
import numpy as np

from stn import STNet, stn_train, stn_test
from stem import Stem
from visualise import imshow, save, plot_pores
from texture import Texture
from minutiae import Minutiae1a, Minutiae1b

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

class TextureNet(nn.Module):
    def __init__(self):
        super(TextureNet, self).__init__()


  This file has been truncated. show original