利用kaggle+pytorch进行机器学习1(图像分类)
最近在学习机器学习,一开始准备适应tensorflow框架,结果学习的时候发现tensrflow2.0与1.0版本兼容性太差,于是采用pytorch框架。 对于学生党而言,免费的GPU是最重要的资源,kaggle提供了每周30个小时的免费GPU,基本满足了学习需要。 没有自己写源码,参考了kaggle上面的一个资源,并作了一定注释和修改。 #导入必要的包 import numpy as np # linear algebra import pandas as pd # data processing, CSV file I/O (e.g. pd.read_csv) import os impo
用户评论