How to measure total training memory?

tianzhi0549 / FCOS

FCOS: Fully Convolutional One-Stage Object Detection (ICCV'19)

https://arxiv.org/abs/1904.01355

Other

3.28k stars 630 forks source link

Closed stigma0617 closed 5 years ago

stigma0617 commented 5 years ago

Hi,

I'm impressed by your good work.

I have a question:

How to measure total training memory you said in README file?

by nvidia-smi or torch.cuda.max_memory_allocated()??

tianzhi0549 commented 5 years ago

@stigma0617 we use torch.cuda.max_memory_allocated.