-
Notifications
You must be signed in to change notification settings - Fork 11
/
opts.lua
33 lines (26 loc) · 1.16 KB
/
opts.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
--------------------------------------------------------------------------------
-- Contains options required by run.lua
--
-- Written by: Abhishek Chaurasia
-- Dated: 6th June, 2016
--------------------------------------------------------------------------------
local opts = {}
lapp = require 'pl.lapp'
function opts.parse(arg)
local opt = lapp [[
Command line options:
-m, --model (default '') Path & filename of network model to profile
-p, --platform (default cpu) Select profiling platform (cpu|cuda)
-r, --res (default 1x3x231x231) Input image resolution Channel x Width x Height
-e, --eye (default 0) Network eye
-i, --iter (default 1000) Averaging iterations
-s, --save (default -) Save the float model to file as <model.net.ascii>in
[a]scii or as <model.net> in [b]inary format (a|b)
--verbose (default detail) detail/medium/compact
--MACs Use multiply-add when counting ops
Example:
th profile-model.lua --model <'path/filename.lua'> --res 1x3x231x231
]]
return opt
end
return opts