Closed zahidemon closed 1 year ago
To test the added variants in the DeepSSM use case, please make the following changes in the config file:
config["loss"]["function"]
parameter as the loss function to train the model. Available options are: MSE
and Focal.
Default value: MSE
config["tl_net"]["enabled"]
parameter as True to train the model using TL-Net. If the value is False, the model will be trained with Vanilla/Base DeepSSM. Default value: False
@akenmorris @jadie1
@jadie1 , any comments?
1881
Main changes: