RetinaNet 在训练过程冻结了BN层 (2), (3),因为 batch size 太小了,这时候使用 BN 学习到的参数难以稳定 在mmdetection 中,大部分 detector 的配置文件中,通过设置 norm_eval=True 使得在训练过程冻结 BN 层,而在推理过程时使用 BN 层。 在训练过程冻结 BN 层是什么意思? 网络不会更新 γγ 和ββ 参数...
它包括base_batch_size和enable。base_batch_size是优化器 lr 所基于的批量大小。enable是打开和关闭该功能的开关。 optim_wrapper– 计算模型参数的梯度。 如果指定,还应指定 train_dataloader。 如果需要自动混合精度或梯度累积训练。optim_wrapper的类型应该是 AmpOptimizerWrapper。 有关示例,请参见 build_optim_wra...
batch_data_samples: OptSampleList = None) -> Tuple[Dict]: #计算输入特征的批量大小(batch size) batch_size = mlvl_feats[0].size(0) #从 batch_data_samples 中构建二进制掩码 masks,以确定图像中的哪些位置将被忽略。. assert batch_data_samples is not None batch_input_shape = batch_data_sampl...
nn.functional.mse_loss(y_pred, y, reduce = True, size_average = True) 很多的 loss 函数都有 size_average 和 reduce 两个布尔类型的参数。因为一般损失函数都是直接计算 batch 的数据,因此返回的 loss 结果都是维度为 (batch_size, ) 的向量。 如果reduce = False,那么 size_average 参数失效,直接返...
DataLoader(dataset,batch_size=1,shuffle=False,sampler=None,batch_sampler=None,num_workers=0,collate_fn=None,pin_memory=False,drop_last=False,timeout=0,worker_init_fn=None,*,prefetch_factor=2,persistent_workers=False) 1 2 3 4 5 num_worker定义了用于load数据的子进程,非0的时候,模型在跑data的...
input_sp_tensor=SparseConvTensor(voxel_features,coors,self.sparse_shape,batch_size)...spatial_features=out.dense() 卷积完,输出为[2, 128, 2, 200, 176],然后height compression(openpcdet中的叫法,其实就是把高度和channels维度给view到了一个维度,全都保存了)得到[2, 256, 200, 176],也就是本步...
(batch_size, out_boxes)).to(scores.device)) /home/group5/anaconda3/envs/SSPNet/lib/python3.7/site-packages/torch/onnx/symbolic_opset9.py:2819: UserWarning: Exporting aten::index operator of advanced indexing in opset 11 is achieved by combination of multiple ONNX operators, including Reshape...
batch_size = mlvl_feats[0].size(0) input_img_h, input_img_w = img_metas[0]['batch_input_shape'] # 创建一个全1的尺寸为pad后图像的mask矩阵 img_masks = mlvl_feats[0].new_ones((batch_size, input_img_h, input_img_w))
samples_per_gpu=1, # 这是batch_size workers_per_gpu=3, # 这是num workers train=dict( type='RepeatDataset', times=1, dataset=dict( type='TileDataset', ann_file=[ data_root + 'train.txt' ], img_prefix=[data_root], pipeline=train_pipeline)), ...
assert batch_size == 1 2022-05-13 09:34:47,130 - mmdeploy - INFO - torch2onnx success. 2022-05-13 09:34:47,132 - mmdeploy - INFO - onnx2tensorrt of /data/pointcloud_alg/mmdeploy/work_dir/end2end.onnx start. 2022-05-13 09:34:47,992 - mmdeploy - INFO - Successfully ...