最近一个任务需要用到Deformable-DETR,其中比较关键的一步是Multi-scale Deformable Attention Module (MSDeformAttn),这个模块 …
read moreOther articles
在NCCL后端下Pytorch的distributed.all_gather卡死排查
用了Github上一个SimCLR的PyTorch实现,但是在训练过程中遇到了一些问题。
原repo要用DDP训练的方式有点质朴,需要手动启动N的进程进行训练,那8张卡岂不是要操作八次 …
read more在Pelican中添加Gitalk
在PyTorch DistributedSampler中封装其他Sampler策略
SIFT算法及Python实现
用Python实现了一遍SIFT算法
read more