Possibly unintended behaviour

[jmg049]

Hi,

I was going through the code base and noticed some potentially unintended behaviour during the training of the MOSEI models. Please see the following lines of code . The if statement sets the set_name "trn" to "train" and then further down in the __init__ method and the __getitem__ method the if conditions check for whether or not the set_name is "trn" and not "train" resulting in the validation and tests blocks being run every time.

[jmg049]

It doesn't impact the __getitem__ condition but it does impact the creation of the missing index in the __init__ method