search

Waikato / wekaDeeplearning4j

Weka package for the Deeplearning4j java library
https://deeplearning.cms.waikato.ac.nz/
GNU General Public License v3.0
185 stars 202 forks source link

How to modify Output Layer in Pretrained Models such as Dl4JResNet50 #63

Closed trainingemad closed 3 years ago

trainingemad commented 3 years ago

Description

I want to know how can we amend Output Layer methods such as activation function, loss function for pre-trained models such as Dl4JResNet50 using Java API.

Expected behavior:

Encountered behavior:

Additional Information