Skip to content

grammar issue in FastCSR #256

Description

@yangwl8

https://github.com/pBFSLab/DeepPrep/blob/04af8f3541737505de41530b23804cd9f2b6efa1/deepprep/FastCSR/fastcsr_model_infer.py#L57
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

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions