7 lines
157 B
Python
7 lines
157 B
Python
|
import os
|
||
|
|
import torch
|
||
|
|
detect_model = torch.load( "/home/ykn/algorithm_system/flask_web/weight/traffic/detect.pt")['model']
|
||
|
|
|
||
|
|
print(detect_model.children())
|
||
|
|
|