JankowskiDaniel / Neural-Deep-Retina

0 stars 0 forks source link

Correct input shape of the encoder #21

Closed kapiblue closed 3 months ago

kapiblue commented 3 months ago

The input shape in load_model.py should take sequence length into consideration:

seq_len = config.data.seq_len
    if seq_len > 1:
        input_shape = (batch_size, seq_len, *img_shape)
    else:
        input_shape = (batch_size, *img_shape)