Closed kbali1297 closed 1 year ago
This seems reasonable. Could you describe the new kwargs
parameter in the documentation notebook?
It should probably just say that this is a placeholder for arguments that are not supported by the function, such as layers
.
Sure, I have added the description for **kwargs in the documentation notebook.
Removed layers default argument in conv_net and res_net and added default argument layers=[] in the CouplingLayer class. Added kwargs as argument in u_net, conv_net and res_net to accommodate the missing layers argument in u_net function definition analogous to layers** in conv_net and res_net function definitions.