Amin Ghiasi

icon indicating a website link cs.umd.edu/~amin icon indicating an email [email protected]

icon company University of Maryland icon location College Park

Results 1 issues of Amin Ghiasi

Does avg_batch work?

https://github.com/yjlolo/vae-audio/blob/3a43e3122da55daf89fffaf5bbf11fdb805a597f/model/loss.py#L31 It seems to me taking `torch.sum(tensor)` outputs a number, so taking: `if avg_batch: output = torch.mean(output, dim=0) ` becomes obsolete. Am I missing something?