Revision d892d1f549bb03e3ee46022678cface22bd6e860 authored by Gregory Chanan on 06 October 2016, 22:49:33 UTC, committed by Gregory Chanan on 08 November 2016, 21:01:05 UTC
1 parent 09acb86
Raw File
init.lua
require "cutorch"
require "nn"
require "cunn.THCUNN"

require('cunn.test')
require('cunn.DataParallelTable')

nn.Module._flattenTensorBuffer['torch.CudaTensor'] = torch.FloatTensor.new
back to top