Meleager comments

Repositories
Issues
Comments

Results 1 comments of


                                            Meleager

errors occur when loading the pretrained model

I have the same problem. The way I tried: Change this part to: https://github.com/DK-Jang/motion_puzzle/blob/52af967040c42c5eb37c48b03d412e81e7b37def/trainer.py#L34-L35 ``` self.gen = self.gen.to(self.device) self.gen_ema = self.gen_ema.to(self.device) ``` And: https://github.com/DK-Jang/motion_puzzle/blob/52af967040c42c5eb37c48b03d412e81e7b37def/trainer.py#L162 `state_dict = torch.load(model_path, map_location="cuda:0")` And do...