Allen Liu

icon company aws ai icon location California

Results 6 comments of Allen Liu

Error training for x2

> RuntimeError: Error(s) in loading state_dict for RRDBNet: > untimeError: Error(s) in loading state_dict for RRDBNet: > Unexpected key(s) in state_dict: "upconv2.weight", "upconv2.bias". Did this solve your problem? If you...

fix: make non-leader nodes sleep rather than wait on orted process for mpi based jobs

In the long run, I suggest we could use a light-weighted database to store status information. For now this PR looks good.

Update the Hook callback to be compatible with xgboost>1.3.0 callback style

Issue should be fixed by PR:https://github.com/awslabs/sagemaker-debugger/pull/648 Closing.

network.py

In VGG class.

network.py

thanks!

module.register_forward_hook only captures some layers when using torch.compile

any update? @wconstab I also got similar error while using Mnist training script with hook: I would be great if you can also provide a ETA of fixing the basic...