设置清华安装源
self.relu = nn.ReLU(inplace=True)其中inplace=Truemeansthat it will modify the input directly, without allocating any additional output. It can sometimes slightly decrease the memory usage, but may not always be a valid operation (because the original inpu.
复制链接