|
@@ -75,7 +75,7 @@ cmd:option("-learning_rate_decay", 3.0e-7, 'learning rate decay (learning_rate *
|
|
|
cmd:option("-resume", "", 'resume model file')
|
|
|
cmd:option("-name", "user", 'model name for user method')
|
|
|
cmd:option("-gpu", 1, 'Device ID')
|
|
|
-cmd:option("-loss", "huber", 'loss function (huber|l1)')
|
|
|
+cmd:option("-loss", "huber", 'loss function (huber|l1|mse)')
|
|
|
|
|
|
local function to_bool(settings, name)
|
|
|
if settings[name] == 1 then
|