for me same error happens and I have the same size as which I had run same program with different lr and initialization properly, No idea why it occurs and what the source of error is
RuntimeError: after cudaLaunch in triple_chevron_launcher::launch(): too many resources requested for launch
Thanks,