-
Notifications
You must be signed in to change notification settings - Fork 19
/
init.lua
38 lines (33 loc) · 911 Bytes
/
init.lua
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
require 'paths'
require 'xlua'
require 'torchx'
require 'string'
require 'os'
require 'sys'
require 'image'
require 'lfs'
-- these actually return local variables but we will re-require them
-- when needed. This is just to make sure they are loaded.
require 'moses'
require 'ffi'
unpack = unpack or table.unpack
local dl = require 'dataload._env'
require 'dataload.config'
require 'dataload.utils'
require 'dataload.buildBigrams'
require 'dataload.DataLoader'
require 'dataload.TensorLoader'
require 'dataload.ImageClass'
require 'dataload.AsyncIterator'
require 'dataload.SequenceLoader'
require 'dataload.MultiSequence'
require 'dataload.MultiImageSequence'
require 'dataload.loadCIFAR10'
require 'dataload.loadMNIST'
require 'dataload.loadPTB'
require 'dataload.loadSentiment140'
require 'dataload.loadImageNet'
require 'dataload.loadGBW'
require 'dataload.preprocess'
require 'dataload.test'
return dl