Skip to content

Latest commit

 

History

History
20 lines (13 loc) · 658 Bytes

PreTrainedModel.md

File metadata and controls

20 lines (13 loc) · 658 Bytes

Link to model trained using PS dataset

The model is built using using nn package from Torch 7

Usage

The patches should be of type CudaTensor of size 1 x 32 x 32 with values ranging from 0-1. Each patch should have zero mean and unit standard deviation, i.e. p:add(-p:mean()) and then p:div(p:std())

require 'torch'
require 'nn'
require 'cudnn'
require 'cunn'

network:evaluate()
descriptor = network:forward(p)

Link to model trained using PS dataset in PyTorch