用了Github上一个SimCLR的PyTorch实现,但是在训练过程中遇到了一些问题。
原repo要用DDP训练的方式有点质朴,需要手动启动N的进程进行训练,那8张卡岂不是要操作八次 …
read more用了Github上一个SimCLR的PyTorch实现,但是在训练过程中遇到了一些问题。
原repo要用DDP训练的方式有点质朴,需要手动启动N的进程进行训练,那8张卡岂不是要操作八次 …
read moreSummary => 在改写deeplabv3+时, 因为需要去掉ASPP和decoder部分, 仅保留mobilenet的backbone. 但是因为误对nn.Module
内的所有权 …