模型非常小却爆显存?


#1

Init stgat0_dense0_weight (8, 3) with Xavier
Init stgat0_dense1_weight (1, 16) with Xavier
Init stgat0_weight (128, 8, 1, 3) with Xavier
Init stgat0__weight (64, 3, 1, 1) with Xavier
Init stgat0_weight (12, 4, 1, 64) with Xavier
Init stgat0_W (307, 12) with Xavier
Init stgat0_weight (8, 3) with Xavier
Init stgat0_weight (1, 16) with Xavier
Init stgat0__weight (128, 8, 1, 3) with Xavier
Init stgat0_weight (64, 3, 1, 1) with Xavier
Init stgat0__weight (12, 4, 1, 64) with Xavier
Init stgat0__W (307, 12) with Xavier
Init stgat0-dense0_weight (8, 3) with Xavier
Init stgat0_dense1_weight (1, 16) with Xavier
Init stgat0_stgat_block2_conv0_weight (128, 8, 1, 3) with Xavier
Init stgat0_stgat_block2_conv1_weight (64, 3, 1, 1) with Xavier
Init stgat0_stgat_submodule2_conv0_weight (12, 42, 1, 64) with Xavier
Init stgat0_stgat_submodule2_W (307, 12) with Xavier

就这么一点参数,24g显存一训练直接爆了?