megengine.utils.network.Network.reset_batch_size

Network.reset_batch_size(batchsize, *, blacklist=())

重置批大小。

参数

blacklist – 不改变批大小的变量集合，通常意味着他们的第一个维度不是批大小。

上一页

megengine.utils.network.Network.replace_vars

下一页

megengine.utils.network.as_varnode