Skip to content

Commit b34b941

Browse files
committed
Update Segmentor.h
1 parent 4005842 commit b34b941

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

include/Segmentor.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -172,7 +172,7 @@ void Segmentor<Model>::Train(float learning_rate, int epochs, int batch_size,
172172
template <class Model>
173173
void Segmentor<Model>::LoadWeight(std::string weight_path){
174174
torch::load(model, weight_path);
175-
model->to(device);
175+
model->to(device);
176176
model->eval();
177177
return;
178178
}

0 commit comments

Comments
 (0)