Block a user
Training with custom dataset
I did the changes in your commit and got rid of the problem, but now I think I made a mistake on the dataset preparation part.
` File "/home/vall_e/utils/trainer.py", line 213, in train …