utils.py File Reference

Go to the source code of this file.

Namespaces
namespace	smrunner
namespace	smrunner.utils

Functions
	smrunner.utils.load_model (model, path, ignore_key=None, device='cuda:0')
	smrunner.utils.save_model (model, save_path)
	smrunner.utils.load_optimizer (optimizer, path)
	smrunner.utils.save_optimizer (optimizer, save_path)
	smrunner.utils.init_weights (model)