(self, opt)
| 76 | |
| 77 | class InferenceAgent: |
| 78 | def __init__(self, opt): |
| 79 | torch.cuda.empty_cache() |
| 80 | self.opt = opt |
| 81 | self.rank = opt.rank |
| 82 | |
| 83 | # Load Model |
| 84 | self.load_model() |
| 85 | self.load_weight(opt.ckpt_path, rank=self.rank) |
| 86 | self.G.to(self.rank) |
| 87 | self.G.eval() |
| 88 | |
| 89 | # Load Data Processor |
| 90 | self.data_processor = DataProcessor(opt) |
| 91 | |
| 92 | def load_model(self) -> None: |
| 93 | self.G = FLOAT(self.opt) |
nothing calls this directly
no test coverage detected