ACmix
ACmix copied to clipboard
已解决: RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.cuda.HalfTensor) should be the same
在position coding时,输入固定float,在yolo训练val时会自动进行半精度训练,将position()输出类型转换成input就好 position(......).to(x.dtype)
哥们,问一哈,可以搞到自然语言处理用吗?
哥们,问一哈,可以搞到自然语言处理用吗?
不知道呀,我是目标检测
Good Job!