逐层模式打印的信息
Model Summary: 213 layers, 7225885 parameters, 0 gradients
Model(
(model): Sequential(
(0): Conv(
(conv): MLUConv2d(
3, 32, kernel_size=(6, 6), stride=(2, 2), padding=(2, 2), scale=Parameter containing:
tensor([127.00000, 4.45803]), quantized_mode=Parameter containing:
tensor([1], dtype=torch.int32), input_mean=Parameter containing:
tensor([0., 0., 0.]), input_std=Parameter containing:
tensor([255., 255., 255.])
)
(act): SiLU()
)
(1): Conv(
(conv): MLUConv2d(
32, 64, kernel_size=(3, 3), stride=(2, 2), padding=(1, 1), scale=Parameter containing:
tensor([ 2.11397, 119.14012]), quantized_mode=Parameter containing:
tensor([1], dtype=torch.int32), input_mean=None, input_std=None
)