うちは「color transfer for src faceset」をnone以外にすると同じような症状になる
他のbatch size等の組み合わせでエラーメッセージが出なかったり、MemoryErrorやスタックトレースが出たりもする
Colabなら問題なくTrain始まるんだけど・・・