mirror of
https://github.com/huggingface/candle.git
synced 2025-06-14 09:57:10 +00:00
11c7e7bd67
* Some fixes for yolo-v3. * Use the running stats for inference in the batch-norm layer. * Get some proper predictions for yolo. * Avoid the quadratic insertion.
8 lines
237 B
Python
8 lines
237 B
Python
def remove_prefix(text, prefix):
|
|
return text[text.startswith(prefix) and len(prefix):]
|
|
nps = {}
|
|
for k, v in model.state_dict().items():
|
|
k = remove_prefix(k, 'module_list.')
|
|
nps[k] = v.detach().numpy()
|
|
np.savez('yolo-v3.ot', **nps)
|