摘要:
讲解 对比学习论文中出现: # compute logits # Einstein sum is more intuitive # positive logits: Nx1 l_pos = torch.einsum('nc,nc->n', [q, k]).unsqueeze(-1) # negati 阅读全文
摘要:
取参数: g=model.named_parameters() parm={} for name,parameters in model.named_parameters(): print(name,':',parameters.size()) parm[name]=parameters.detac 阅读全文
摘要:
parser.add_argument("--flag", action="store_true", help="Run or not.") 当你不输入 --flag 的时候,默认为 False;输入 --flag 的时候,才会触发 True 值。【符合常理】 当然用方式,反过来也可以指定 acti 阅读全文