SinGAN(ICCV 2019)程序问题

Traceback (most recent call last): File “D:/python2018/SinGAN/main_train.py”, line 2, in from SinGAN.manipulate import * File “D:\python2018\SinGAN\SinGAN\manipulate.py”, line 11, in import torchvision.datasets as dset File “D:\anaconda3\lib\site-packages\torchvision_init_.py”, line 1, in from torchvision import models File “D:\anaconda3\lib\site-packages\torchvision\models_init_.py”, line 12, in from . import detection File “D:\anaconda3\lib\site-packages\torchvision\models\detection_init_.py”, line 1, in from .faster_rcnn import * File “D:\anaconda3\lib\site-packages\torchvision\models\detection\faster_rcnn.py”, line 13, in from .rpn import AnchorGenerator, RPNHead, RegionProposalNetwork File “D:\anaconda3\lib\site-packages\torchvision\models\detection\rpn.py”, line 8, in from . import utils as det_utils File “D:\anaconda3\lib\site-packages\torchvision\models\detection_utils.py”, line 74, in @torch.jit.script File "D:\anaconda3\lib\site-packages\torch\jit_init.py", line 364, in script graph = _script_graph(fn, _frames_up=frames_up + 1) File "D:\anaconda3\lib\site-packages\torch\jit_init.py", line 359, in _script_graph ast = get_jit_ast(fn) File “D:\anaconda3\lib\site-packages\torch\jit\frontend.py”, line 132, in get_jit_ast return build_def(SourceRangeFactory(source), py_ast.body[0]) File “D:\anaconda3\lib\site-packages\torch\jit\frontend.py”, line 151, in build_def build_stmts(ctx, body)) File “D:\anaconda3\lib\site-packages\torch\jit\frontend.py”, line 123, in build_stmts stmts = [build_stmt(ctx, s) for s in stmts] File “D:\anaconda3\lib\site-packages\torch\jit\frontend.py”, line 123, in stmts = [build_stmt(ctx, s) for s in stmts] File “D:\anaconda3\lib\site-packages\torch\jit\frontend.py”, line 140, in call return method(ctx, node) File “D:\anaconda3\lib\site-packages\torch\jit\frontend.py”, line 205, in build_Assign rhs = build_expr(ctx, stmt.value) File “D:\anaconda3\lib\site-packages\torch\jit\frontend.py”, line 140, in call return method(ctx, node) File “D:\anaconda3\lib\site-packages\torch\jit\frontend.py”, line 314, in build_Call func = build_expr(ctx, expr.func) File “D:\anaconda3\lib\site-packages\torch\jit\frontend.py”, line 140, in call return method(ctx, node) File “D:\anaconda3\lib\site-packages\torch\jit\frontend.py”, line 300, in build_Attribute value = build_expr(ctx, expr.value) File “D:\anaconda3\lib\site-packages\torch\jit\frontend.py”, line 140, in call return method(ctx, node) File “D:\anaconda3\lib\site-packages\torch\jit\frontend.py”, line 422, in build_Subscript raise NotSupportedError(base.range(), “slicing multiple dimensions at the same time isn’t supported yet”) torch.jit.frontend.NotSupportedError: slicing multiple dimensions at the same time isn’t supported yet proposals (Tensor): boxes to be encoded “”"

# perform some unpacking to make it JIT-fusion friendly wx = weights[0] wy = weights[1] ww = weights[2] wh = weights[3] proposals_x1 = proposals[:, 0].unsqueeze(1) ~~~~~~~~~ <--- HERE proposals_y1 = proposals[:, 1].unsqueeze(1) proposals_x2 = proposals[:, 2].unsqueeze(1) proposals_y2 = proposals[:, 3].unsqueeze(1) reference_boxes_x1 = reference_boxes[:, 0].unsqueeze(1) reference_boxes_y1 = reference_boxes[:, 1].unsqueeze(1) reference_boxes_x2 = reference_boxes[:, 2].unsqueeze(1) reference_boxes_y2 = reference_boxes[:, 3].unsqueeze(1)