专栏名称: 爱可可-爱生活
知名互联网资讯博主 北邮PRIS模式识别实验室陈老师
目录
相关文章推荐
新智元  ·  AI卷翻科研!DeepMind ... ·  昨天  
黄建同学  ·  玩转AI游戏开发!#ai##科技# ... ·  6 天前  
51好读  ›  专栏  ›  爱可可-爱生活

【nano-simsiam:一个极简的PyTorch实现的Sim-20241127130737

爱可可-爱生活  · 微博  · AI  · 2024-11-27 13:07

正文

2024-11-27 13:07

【nano-simsiam:一个极简的PyTorch实现的SimSiam自监督学习框架,仅用400行代码实现,基于ResNet50在ImageNet上达到良好性能。支持分布式训练、实时KNN评估和自动混合精度训练,适合研究原型开发,代码简洁易于理解和修改】
'Minimalistic, hackable PyTorch implementation of SimSiam in ~400 lines. Achieves good performance on ImageNet with ResNet50. Features distributed training, real-time KNN eval, and AMP.'
GitHub: github.com/lucas-maes/nano-simsiam
#自监督学习# #PyTorch# #深度学习# #计算机视觉#