luxianlebron

Results 3 issues of luxianlebron

我每次训练次数达到15次,以后的训练自动都是初始化参数训进行训练,如何修改代码?

**Describe the bug** When I use an onnx model to convert to pb, the following error appears: ![image](https://user-images.githubusercontent.com/82278841/122345197-2c3bac00-cf7a-11eb-81d6-78f866fe5f30.png) **To Reproduce** This is the code I used when converting: ![image](https://user-images.githubusercontent.com/82278841/122345326-4f665b80-cf7a-11eb-90e0-03e1274530e9.png) The...

Thank you for your work. gmm = {k: gmm[k].reshape(1, -1).expand(pred.shape[0], -1) for k in gmm} mse_term = F.mse_loss(pred, target, reduction='none') / 2 / noise_var + 0.5 * noise_var.log() sum_var =...