add `load_strict_model_weights` as an optional config parameter (#655) #244

Job	Run time
docker-build (1.13.1_cu117, mosaicml/pytorch:1.13.1_cu117-python3.10-ubuntu20.04, [gpu])	6m 16s
docker-build (2.0.1_cu118, mosaicml/pytorch:2.0.1_cu118-python3.10-ubuntu20.04, [gpu])	1m 22s
docker-build (2.1.0_cu121, mosaicml/pytorch:2.1.0_cu121-python3.10-ubuntu20.04, [gpu])	1m 30s
docker-build (2.1.0_cu121_flash2, mosaicml/pytorch:2.1.0_cu121-python3.10-ubuntu20.04, [gpu-flash2])	1m 33s
	10m 41s

Provide feedback