-
Notifications
You must be signed in to change notification settings - Fork 21
/
Copy pathdata_threads.lua
32 lines (32 loc) · 888 Bytes
/
data_threads.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
print('run loader threads')
threads = require('threads')
data_loader = dofile('data_loader.lua')
do
local options = opt
local train = training_data
local val = validation_data
local video_train = video_training_data
local video_val = video_validation_data
local video_test = video_test_data
task_queue = threads.Threads(
opt.n_threads,
function()
require('torch')
require('image')
dofile('mean.lua')
utils = dofile('utils.lua')
data_loader = dofile('data_loader.lua')
end,
function(thread_id)
opt = options
id = thread_id
local seed = opt.manual_seed + id
torch.manualSeed(seed)
training_data = train
validation_data = val
video_training_data = video_train
video_validation_data = video_val
video_test_data = video_test
end
)
end