search

XiYe20 / VPTR

The repository for paper VPTR: Efficient Transformers for Video Prediction
MIT License
88 stars 19 forks source link

code for testing BAIR dataset #11

Closed JunyaoHu closed 1 year ago

JunyaoHu commented 1 year ago

I use your Test_VPTR.ipynb to test on BAIR dataset, but I get a wrong output which is more and more gray.

ground truth

image

result

image

Test_VPTR_BAIR_my.txt

XiYe20 commented 1 year ago

Hi, did you set the decoder outlayer to be 'Tanh' for BAIR dataset?

JunyaoHu commented 1 year ago

it works thanks! i think you can update the comment here, i didnt notice your train file

image

https://github.com/XiYe20/VPTR/blob/b876364ee19100dccde35ef402bcc2fb1930fdf1/train_FAR.py#L180

image