in train.py file: ``` if not args.lstm: data = model.core_hh.W.data data[:] = np.identity(data.shape[0], dtype=np.float32) ```
in train.py file: