我想把优化器替换为RMSprop,看了basicsr的源码也有实现RMSprop优化器方式,optim_type == 'RMSprop':
optimizer = torch.optim.RMSprop(params, lr, **kwargs)。
但是为什么会出现没有实现的错误:basicsr/models/base_model.py", line 107, in get_optimizer
raise NotImplementedError(f'optimizer {optim_type} is not supperted yet.')
NotImplementedError: optimizer RMSprop is not supperted yet.
试了一下,配置参数只有写Adam才会有效
我想把优化器替换为RMSprop,看了basicsr的源码也有实现RMSprop优化器方式,optim_type == 'RMSprop': optimizer = torch.optim.RMSprop(params, lr, **kwargs)。 但是为什么会出现没有实现的错误:basicsr/models/base_model.py", line 107, in get_optimizer raise NotImplementedError(f'optimizer {optim_type} is not supperted yet.') NotImplementedError: optimizer RMSprop is not supperted yet. 试了一下,配置参数只有写Adam才会有效