i'm calling .cuda() in a few spots assuming cuda is available.
i'm calling .cuda() in a few spots assuming cuda is available.