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())
|
|
|