problem with alphafold2/train_pre.py

[jackli777]

here is the problem when running alphafold2/train_pre.py:

/usr/local/lib/python3.7/dist-packages/torch/_tensor.py:575: UserWarning: floor_divide is deprecated, and will be removed in a future version of pytorch. It currently rounds toward 0 (like the 'trunc' function NOT 'floor'). This results in incorrect rounding for negative values.
To keep the current behavior, use torch.div(a, b, rounding_mode='trunc'), or for actual floor division, use torch.div(a, b, rounding_mode='floor'). (Triggered internally at /pytorch/aten/src/ATen/native/BinaryOps.cpp:467.)
return torch.floor_divide(self, other)

RuntimeError Traceback (most recent call last)
in ()
114 distogram,
115 discretized_distances,
--> 116 ignore_index = IGNORE_INDEX
117 )
118

/usr/local/lib/python3.7/dist-packages/torch/nn/functional.py in cross_entropy(input, target, weight, size_average, ignore_index, reduce, reduction)
2822 if size_average is not None or reduce is not None:
2823 reduction = _Reduction.legacy_get_string(size_average, reduce)
-> 2824 return torch._C._nn.cross_entropy_loss(input, target, weight, _Reduction.get_enum(reduction), ignore_index)
2825
2826

RuntimeError: input and target batch or spatial sizes don't match: target [1 x 52 x 52], input [1 x 256 x 156 x 156] at /pytorch/aten/src/THCUNN/generic/SpatialClassNLLCriterion.cu:23