if __name__ == '__main__': module = CAM_Module() in_data = torch.randint(0, 255, (2, 3, 7, 7), dtype=torch.float32)
如何自动生成图片用于测试 pytorch(No image? No need image)
猜你喜欢
转载自www.cnblogs.com/ziytong/p/10686055.html
今日推荐
周排行