annotated-transformer
annotated-transformer copied to clipboard
Epoch Training: Help
Not sure, what is wrong? Any suggestions
/usr/local/lib/python3.7/dist-packages/torch/nn/reduction.py:42: UserWarning: size_average and reduce args will be deprecated, please use reduction='sum' instead. warnings.warn(warning.format(ret)) /usr/local/lib/python3.7/dist-packages/ipykernel_launcher.py:20: UserWarning: nn.init.xavier_uniform is now deprecated in favor of nn.init.xavier_uniform.
TypeError Traceback (most recent call last)
10 frames
TypeError: mean() received an invalid combination of arguments - got (int, keepdmin=bool), but expected one of:
- (*, torch.dtype dtype) didn't match because some of the keywords were incorrect: keepdmin
- (tuple of ints dim, bool keepdim, *, torch.dtype dtype)
- (tuple of names dim, bool keepdim, *, torch.dtype dtype)
----> 9 mean = x.mean(-1, keepdmin=True)
keepdmin is a typo, the correct argument is keepdim