detach (1) [PyTorch] .detach().cpu().numpy()와 .cpu().data.numpy() ? PyTorch를 사용하다보면, Module을 통해 나온 Tensor을 후처리에 사용하거나, 계산된 loss Tensor을 logging 하는 일이 많다. for i, data in enumerate(dataloader): # 생략 embeddings = model(images) # total_embeddings에 embeddings append # total_embeddings.append(embeddings) ??? 이때 embeddings는 GPU에 올라가있는 Tensor 이기 때문에 numpy 혹은 list로의 변환이 필요하다. 오픈소스를 보면 detach(), cpu(), data, numpy(), tolist() 등을 조합해서 변환을 한다. 하지만 stackoverflow나 pytorch dis.. 이전 1 다음
