could you please offer a tarined model for this code? It's too slow for my 3090

[awanogawa]

I try to use my 1066 and 3090 to train this model, but I found there is no difference between the two devices, I think the problem may be the structure of the creation of the neural network. could you please offer a tarined model for this code? Thanks very much.

[ying-doc]

Have you solved this problem, please? About solving the problem of too slow training.

[Hanjieee923]

After I replace this code with my own data set, my reward does not converge. Have you ever been in this situation?