运行github中NeRF-W代码时出现的问题和解决方法

GitHub地址:GitHub - kwea123/nerf_pl at nerfw

问题1:

运行以下代码时出现错误,报错信息为:

class NeRFSystem(LightningModule):
    def __init__(self, hparams):
        super().__init__()
        self.hparams = hparams
        ···
    ···
···

报错信息为: 
运行github中NeRF-W代码时出现的问题和解决方法_第1张图片


这可能是因为版本问题,解决方法为改为以下其一(可以都试试,我只有第三种能行):

  • for key in hparams.keys():
             self.hparams[key]=hparams[key]
  • self.hparams.update(hparams)
  • self.save_hyperparameters(hparams)

问题2

def main(hparams):
    system = NeRFSystem(hparams)
    checkpoint_callback = \
        ModelCheckpoint(filepath=os.path.join(f'ckpts/{hparams.exp_name}',
                                               '{epoch:d}'),
                        monitor='val/psnr',
                        mode='max',
                        save_top_k=-1)
···

出现错误:TypeError: __init__() got an unexpected keyword argument 'filepath'

解决方法:依然是版本问题,把filepath改为dirpath即可。

代码跑通了欸!

你可能感兴趣的:(pytorch,深度学习)