generator = torch.Generator("cuda").manual_seed(0)                                                                                                                                                                                            