|
@@ -117,5 +117,5 @@ exporting!"
|
|
|
return {
|
|
return {
|
|
|
"weight_path": self.export_config.weight_path,
|
|
"weight_path": self.export_config.weight_path,
|
|
|
"save_dir": self.global_config.output,
|
|
"save_dir": self.global_config.output,
|
|
|
- "device": self.get_device(),
|
|
|
|
|
|
|
+ "device": self.get_device(1),
|
|
|
}
|
|
}
|