,,,,

from torchstat import stat
stat(model, (1, 30, 30))
posted @ 2022-02-17 11:12  祥瑞哈哈哈  阅读(20)  评论(0)    收藏  举报