t_wの輪郭

ウっ、torch.cuda.OutOfMemoryErrorで死んだ。

torch_dtype=torch.bfloat16つけたらギリ行けた。

1万字つっこもうとしたらtorch.cuda.OutOfMemoryErrorでまた死んだ。