features_in_hook = [] features_out_hook = [] def hook(m… 继续阅读 PyTorch打印指定层的输入输出
月度归档: 2024年3月
Colab保存训练好的模型文件
连接Google Drive from google.colab import drive drive.mou… 继续阅读 Colab保存训练好的模型文件
Llama2 可用地址
https://www.llama2.ai/ https://huggingface.co/chat/
PyCharm设置调式时可预览PyTorch Tensor图像
from PIL import Image from torchvision.transforms impor… 继续阅读 PyCharm设置调式时可预览PyTorch Tensor图像