Commit f6f87aa

committed

Fix RuntimeError: Caught RuntimeError in pin memory thread for device

0." ref #172

1 parent 0572a9e commit f6f87aaCopy full SHA for f6f87aa

File tree

-1

lines changed

-1

lines changed

Lines changed: 3 additions & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -340,8 +340,10 @@ def collate_fn(batch):`
`340`	`340`	`s, e = 1, max_decoder_target_len + 1`
`341`	`341`	`# if b_pad > 0:`
`342`	`342`	`# s, e = s - 1, e - 1`
	`343`	`+ # NOTE: needs clone to supress RuntimeError in dataloarder...`
	`344`	`+ # ref: https://github.com/pytorch/pytorch/issues/10756`
`343`	`345`	`frame_positions = torch.arange(s, e).long().unsqueeze(0).expand(`
`344`		`- len(batch), max_decoder_target_len)`
	`346`	`+ len(batch), max_decoder_target_len).clone()`
`345`	`347`
`346`	`348`	`# done flags`
`347`	`349`	`done = np.array([_pad(np.zeros(len(x[1]) // r // downsample_step - 1),`

Comments

(0)