[https://github.com/pBFSLab/DeepPrep/blob/04af8f3541737505de41530b23804cd9f2b6efa1/deepprep/FastCSR/fastcsr_model_infer.py#L57](url) `network.to(device)` should be `network.to(self.device)` current code format: ``` class LevelsetPredictor(object): def __init__(self, hemi, device, model_path: Path = Path('model')): ... ... self.device = device ... ... # 模型结构创建 def create_network(self): ... ... network.to(device) return network ```
https://github.com/pBFSLab/DeepPrep/blob/04af8f3541737505de41530b23804cd9f2b6efa1/deepprep/FastCSR/fastcsr_model_infer.py#L57
network.to(device)should benetwork.to(self.device)current code format: