2022年2月17日
算模型Params
摘要： summary(model.cuda(), input_size=(3, 256, 256), batch_size=1) 阅读全文
posted @ 2022-02-17 13:08 祥瑞哈哈哈 阅读(30) 评论(0) 推荐(0)
，，，，
摘要： from torchstat import stat stat(model, (1, 30, 30)) 阅读全文
posted @ 2022-02-17 11:12 祥瑞哈哈哈 阅读(20) 评论(0) 推荐(0)
搭建卷积
摘要： x=torch.randn(1, 1, 30, 30) weight=torch.Tensor( 3, 1, *(3,3)) b=torch.Tensor( 1) print(F.conv2d(x,weight,stride=1,padding=0).shape) class CondConv2D( 阅读全文
posted @ 2022-02-17 09:49 祥瑞哈哈哈 阅读(22) 评论(0) 推荐(0)