problem in train code ,embedding = torch.cat((embedding, enc_output), 0) #10

primary-studyer · 2023-02-14T11:54:24Z

embedding = torch.cat((embedding, enc_output), 0)
why cat embedding and enc_output?
and my log has some problem.

log:
torch.Size([1, 64, 64, 64]) embedding value
100%|██████████| 2/2 [00:00<00:00, 2.10it/s]
torch.Size([8, 64, 32, 32]) output for enc_output
why not use enc_output only?

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

problem in train code ,embedding = torch.cat((embedding, enc_output), 0) #10

problem in train code ,embedding = torch.cat((embedding, enc_output), 0) #10

primary-studyer commented Feb 14, 2023

problem in train code ,embedding = torch.cat((embedding, enc_output), 0) #10

problem in train code ,embedding = torch.cat((embedding, enc_output), 0) #10

Comments

primary-studyer commented Feb 14, 2023