pub fn create_sample_model(name: &str, num_layers: usize) -> TorshModel
Create a sample model for testing