search

seominseok0429 / Implicit-Stacked-Autoregressive-Model-for-Video-Prediction

Implicit Stacked Autoregressive Model for Video Prediction (official implementation)
84 stars 6 forks source link

错误的关键字"is_train" #13

Open 2564089649 opened 1 year ago

2564089649 commented 1 year ago

在exp.py文件中，162行，pred_y = self.ema_model(batch_x, pred_list, t, is_train=False) 报错，forword() got an unexpected keyword argument 'is_train'

2564089649 commented 1 year ago

我选择直接去掉is_train参数，然后出现了第二个报错，在metrics.py文件的第32行ssim += cal_ssim(pred[b, f].swapaxes(0, 2), true[b, f].swapaxes(0, 2), multichannel=True)，报错如下：ValueError: win_size exceeds image extent. Either ensure that your images are at least 7x7; or pass win_size explicitly in the function call, with an odd value less than or equal to the smaller side of your images. If your images are multichannel (with color channels), set channel_axis to the axis number corresponding to the channels.