TypeError: nesterov_momentum() got an unexpected keyword argument 'logger'

[xiao751]

When I get to this row, I get an error. Why is that？
"metrics = model.train(train_data, n_epochs=n_epochs, logger=logger, update_fn=update_fn)"
[INFO] Training CoxMLP

TypeError Traceback (most recent call last)
in
18
19 # If you have validation data, you can add it as the second parameter to the function
---> 20 metrics = model.train(train_data, n_epochs=n_epochs, logger=logger, update_fn=update_fn)

D:\anaconda\lib\site-packages\deepsurv\deep_surv.py in train(self, train_data, valid_data, n_epochs, validation_frequency, patience, improvement_threshold, patience_increase, verbose, update_fn, **kwargs)
366 reached, looks at validation improvement to increase patience or
367 early stop.
--> 368 improvement_threshold: percentage of improvement needed to increase
369 patience.
370 patience_increase: multiplier to patience if threshold is reached.

D:\anaconda\lib\site-packages\deepsurv\deep_surv.py in _get_train_valid_fn(self, L1_reg, L2_reg, learning_rate, **kwargs)
208 updates = update_fn(
209 scaled_grads, self.params, **kwargs
--> 210 )
211 else:
212 updates = update_fn(

D:\anaconda\lib\site-packages\deepsurv\deep_surv.py in _get_loss_updates(self, L1_reg, L2_reg, update_fn, max_norm, deterministic, **kwargs)
179 Returns Theano expressions for the network's loss function and parameter
180 updates.
--> 181
182 Parameters:
183 L1_reg: float for L1 weight regularization coefficient.

TypeError: nesterov_momentum() got an unexpected keyword argument 'logger'

[jaredleekatzman]

Are you still having this issue? It looks like the logger is being passed from the .train() function to the _get_train_valid_fn() which is then passing it up update_fn through the **kwargs.

Adding a logger=None parameter to the function signature of _get_train_valid_fn might fix the problem.

For example:

_get_train_valid_fn(self, L1_reg, L2_reg, learning_rate, **kwargs)

Would become:

_get_train_valid_fn(self, L1_reg, L2_reg, learning_rate, logger=None, **kwargs)