yeye
Results
1
comments of
yeye
把这块注释掉就可以了，这一块应该打印出的是gpu的使用情况 train.py 的242 243 244行 # if th.cuda.is_available(): # gpu_mem = cudause.gpu_mem_get(_id=0) # record.log_single(key="gpu_mem", value=gpu_mem)