I’ve thought about this, but when I change the input dimensions, I get hit with this batch size error, which I can’t make heads or tails of:
Traceback (most recent call last):
File “/home/andrew/PyCharm/PycharmProjects/CantStop/Cant-Stop/Python Files/TestNet.py”, line 119, in 
output, loss = train(name_tensor, gender_tensor)
File “/home/andrew/PyCharm/PycharmProjects/CantStop/Cant-Stop/Python Files/TestNet.py”, line 108, in train
loss = loss_func(output, target)
File “/home/andrew/anaconda3/lib/python3.6/site-packages/torch/nn/modules/module.py”, line 206, in call
result = self.forward(*input, **kwargs)
File “/home/andrew/anaconda3/lib/python3.6/site-packages/torch/nn/modules/loss.py”, line 36, in forward
return backend_fn(self.size_average, weight=self.weight)(input, target)
File "/home/andrew/anaconda3/lib/python3.6/site-packages/torch/nn/functions/thnn/auto.py", line 41, in forward
output, *self.additional_args)
RuntimeError: Assertion `THIndexTensor(size)(target, 0) == batch_size’ failed.  at /py/conda-bld/pytorch_1493681908901/work/torch/lib/THNN/generic/ClassNLLCriterion.c:50