Nguyen Bao Hung

icon indicating a website link https://www.linkedin.com/in/bao-hung-nguyen-619aa535/ icon indicating an email [email protected]

icon company TMA Innovation Center icon location Ho Chi Minh, Viet Nam icon location Data Scientist

Results 2 comments of Nguyen Bao Hung

CUDA out of memory error: while allocating the memory

> v = v / np.sqrt(torch.sum(v * v)) replace ` np.sqrt(torch.sum(v * v))` by `v.norm` It worked for me. I think that `np.sqrt()` requires a variable on `cpu`, not `gpu`

CUDA out of memory error: while allocating the memory

I observed that the out-of-memory still occurs even I change batch size to 16. The first round was OK, but the second wasn't. I think we should delete the previous...