pytorch自定义网络层以及损失函数

转自:https://blog.csdn.net/dss_dssssd/article/details/82977170

https://blog.csdn.net/dss_dssssd/article/details/82980222

https://blog.csdn.net/dss_dssssd/article/details/84103834 

 

 

下面是使用squential来自定义网络层的例子:

 

自定义损失函数:

 

 

posted @ 2019-08-19 11:35  rainsoul  Views(2848)  Comments(0Edit  收藏  举报