Colab Issue

[metaphorz]

trafficstars

I am really enjoying experimenting with this! I found a cell error in the Colab notebook, and just ignored it and proceeded with subsequent cells. The cell in question is the one producing the 2nd segmentation. Error:

IndexError Traceback (most recent call last)

in () 2 print("c_code", c_code.shape, c_code.dtype) 3 print("c_indices", c_indices.shape, c_indices.dtype) ----> 4 assert c_code.shape[2]*c_code.shape[3] == c_indices.shape[1] 5 segmentation_rec = model.cond_stage_model.decode(c_code) 6 show_segmentation(torch.softmax(segmentation_rec, dim=1))

IndexError: tuple index out of range

[adeptflax]

I have the same error I'm going to try to figure it out.

[Jiang-maomao]

I also got the same error

[takuya-andou]

I had the same problem, but I changed it to the following and it worked.

not worked assert c_code.shape[2]*c_code.shape[3] == c_indices.shape[1]

worked assert c_code.shape[2]*c_code.shape[3] == c_indices.shape[0]

[metaphorz]

@takuya-andou that does fix it. Thanks! Now, I am wondering about the color coding of segmentation maps (how to choose the proper colors), but that is another topic so I'll research.