Fix typo in val_batch_size and remove unused imports

[debbywehner]

Fix typo in val_batch_size and remove unused imports

Issue #, if available:

N/A

Description of changes:

This pull request makes two changes:

• updates a typo from train_batch_size to val_batch_size, which as an available arg
• remove unused imports

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.