Skip to content

D
dlib
Commit 50143615 authored by Davis King's avatar Davis King
Browse files
Options

minor cleanup

parent e971da97
Hide whitespace changes
Inline Side-by-side
Showing with 16 additions and 8 deletions
dlib/dnn/trainer.h
View file @ 50143615
...@@ -109,12 +109,14 @@ namespace dlib ...@@ -109,12 +109,14 @@ namespace dlib
// and data transfer between the host and the device. // and data transfer between the host and the device.
if (j < data.size()) if (j < data.size())
{ {
net.to_tensor(data.begin()+j, data.begin()+std::min(j+mini_batch_size,data.size()), t1); net.to_tensor(data.begin()+j,
data.begin()+std::min(j+mini_batch_size,data.size()), t1);
j += mini_batch_size; j += mini_batch_size;
} }
if (j < data.size()) if (j < data.size())
{ {
net.to_tensor(data.begin()+j, data.begin()+std::min(j+mini_batch_size,data.size()), t2); net.to_tensor(data.begin()+j,
data.begin()+std::min(j+mini_batch_size,data.size()), t2);
j += mini_batch_size; j += mini_batch_size;
} }
...@@ -125,7 +127,8 @@ namespace dlib ...@@ -125,7 +127,8 @@ namespace dlib
i += mini_batch_size; i += mini_batch_size;
if (j < data.size()) if (j < data.size())
{ {
net.to_tensor(data.begin()+j, data.begin()+std::min(j+mini_batch_size,data.size()), t1); net.to_tensor(data.begin()+j,
data.begin()+std::min(j+mini_batch_size,data.size()), t1);
j += mini_batch_size; j += mini_batch_size;
} }
...@@ -135,7 +138,8 @@ namespace dlib ...@@ -135,7 +138,8 @@ namespace dlib
i += mini_batch_size; i += mini_batch_size;
if (j < data.size()) if (j < data.size())
{ {
net.to_tensor(data.begin()+j, data.begin()+std::min(j+mini_batch_size,data.size()), t2); net.to_tensor(data.begin()+j,
data.begin()+std::min(j+mini_batch_size,data.size()), t2);
j += mini_batch_size; j += mini_batch_size;
} }
} }
...@@ -165,12 +169,14 @@ namespace dlib ...@@ -165,12 +169,14 @@ namespace dlib
// and data transfer between the host and the device. // and data transfer between the host and the device.
if (j < data.size()) if (j < data.size())
{ {
net.to_tensor(data.begin()+j, data.begin()+std::min(j+mini_batch_size,data.size()), t1); net.to_tensor(data.begin()+j,
data.begin()+std::min(j+mini_batch_size,data.size()), t1);
j += mini_batch_size; j += mini_batch_size;
} }
if (j < data.size()) if (j < data.size())
{ {
net.to_tensor(data.begin()+j, data.begin()+std::min(j+mini_batch_size,data.size()), t2); net.to_tensor(data.begin()+j,
data.begin()+std::min(j+mini_batch_size,data.size()), t2);
j += mini_batch_size; j += mini_batch_size;
} }
...@@ -181,7 +187,8 @@ namespace dlib ...@@ -181,7 +187,8 @@ namespace dlib
i += mini_batch_size; i += mini_batch_size;
if (j < data.size()) if (j < data.size())
{ {
net.to_tensor(data.begin()+j, data.begin()+std::min(j+mini_batch_size,data.size()), t1); net.to_tensor(data.begin()+j,
data.begin()+std::min(j+mini_batch_size,data.size()), t1);
j += mini_batch_size; j += mini_batch_size;
} }
...@@ -191,7 +198,8 @@ namespace dlib ...@@ -191,7 +198,8 @@ namespace dlib
i += mini_batch_size; i += mini_batch_size;
if (j < data.size()) if (j < data.size())
{ {
net.to_tensor(data.begin()+j, data.begin()+std::min(j+mini_batch_size,data.size()), t2); net.to_tensor(data.begin()+j,
data.begin()+std::min(j+mini_batch_size,data.size()), t2);
j += mini_batch_size; j += mini_batch_size;
} }
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment