Open kbressem opened 1 year ago
Unless allow_missing_keys is set to true for all base transforms, trainer.predict will throw an error. A temporary overwrite to allow missing keys in all predictions would be useful.
allow_missing_keys
trainer.predict
Unless
allow_missing_keysis set to true for all base transforms,trainer.predictwill throw an error. A temporary overwrite to allow missing keys in all predictions would be useful.