Validation is done using the test set

In model.py, the train method of EfficientCapsNet includes the following line:

[dataset_train, dataset_val = dataset.get_tf_data() ](https://github.com/EscVM/Efficient-CapsNet/blob/4b337f5bf79d70a56e4dec8b113fe54b44cfe963/models/model.py#L158)

However, get_tf_data() simply calls [generate_tf_data()](https://github.com/EscVM/Efficient-CapsNet/blob/4b337f5bf79d70a56e4dec8b113fe54b44cfe963/utils/pre_process_smallnorb.py#L72), which augments and returns the **test dataset**.

Was this done on purpose? My understanding is that the test set should remain unseen until the actual test.

Cheers!

Stijn