generated from OBJNULL/Dockerized-Rust
Fixed type error
This commit is contained in:
parent
8b918113ac
commit
31c30ed200
1 changed files with 1 additions and 1 deletions
|
@ -13,7 +13,7 @@ pub fn get_operation_mode() -> Option<OperationMode> {
|
||||||
let args: Vec<String> = env::args().collect();
|
let args: Vec<String> = env::args().collect();
|
||||||
|
|
||||||
// Getting operation mode
|
// Getting operation mode
|
||||||
match &args[1] {
|
match args[1].as_str() {
|
||||||
"training" => Some(OperationMode::Training),
|
"training" => Some(OperationMode::Training),
|
||||||
"infrence" => Some(OperationMode::Infrence),
|
"infrence" => Some(OperationMode::Infrence),
|
||||||
_ => None,
|
_ => None,
|
||||||
|
|
Loading…
Reference in a new issue