Inform when weights loading promotes types #126000
Labels
module: nn
Related to torch.nn
triaged
This issue has been looked at a team member, and triaged and prioritized into an appropriate module
馃殌 The feature, motivation and pitch
This motivates my issue https://discuss.pytorch.org/t/state-dict-and-loading-of-buffers/202015/2
Here is the problem when you do loading of the weights:
It should not be like this, as it is very hard to diagnose why the loaded model behaves differently now. The message should read like
Alternatives
There maybe needs to be some sort of warning or notification instead of saying "All keys matched successfully", which is clearly not the case
Additional context
No response
cc @albanD @mruberry @jbschlosser @walterddr @mikaylagawarecki
The text was updated successfully, but these errors were encountered: