nsff_pl
nsff_pl copied to clipboard
--hard_sampling has a memory leak
To reproduce, pick a large batch_size and run with hard_sampling. Eventually it will throw RuntimeError: CUDA out of memory