2 Commits

Author	SHA1	Message	Date
Ryan Dick	61242bf86a	Fix bug in skip_torch_weight_init() where the original behavior of torch.nn.Conv*d modules wasn't being restored correctly.	2023-10-10 10:05:50 -04:00
Ryan Dick	58b56e9b1e	Add a skip_torch_weight_init() context manager to improve model load times (from disk).	2023-10-09 14:12:56 -04:00