if gpu:
    network.to("cuda")

i changed it but it doesn’t work
i think it says if gpu is true then run on cuda
so i think i have to change where gpu is true and make it default as false
but i don’t know where is that