Wang Yi
Wang Yi
目前的代码不是最终版本,包括代码所在的文件、代码的实现等,都有待优化。 当前的版本是能跑出初步结果的版本
复现代码 ```python import tempfile import oneflow as flow from oneflow_onnx.oneflow2onnx.util import convert_to_onnx_and_check from flowvision.models.face_recognition import iresnet50 model = iresnet50().to("cuda") class ModelGraph(flow.nn.Graph): def __init__(self, model): super().__init__() self.config.allow_fuse_add_to_output(True) self.backbone = model def...
TODO: - [x] fix bug - [x] doc - [x] unittest
TODO: unittest for SSIM
## Summary by CodeRabbit - **New Features** - Command-line enhancements for image generation, including options to specify data type, model revision, and local file restrictions. - Introduction of a comprehensive...