pennywise86hw

Results 1 issues of pennywise86hw

What are the minimal memory requirements for train_wave_equation.py? I am trying to run it on my gpu with 16GB memory, but consistently getting cuda OOM..