以hw4为例
注册kaggle账号需要tizi来获取验证码,之后登录不需要tizi
使用GPU需要填写手机号,也需要tizi进行验证码验证,每周有37小时的gpu使用时间,也可以使用cpu
1.首先读取训练数据
根据右侧目录结构
修改读取数据集的路径
def parse_args():
"""arguments"""
config = {
#修改读取数据集的路径
"data_dir": "/kaggle/input/ml2022spring-hw4/Dataset",
"save_path": "model.ckpt",
"batch_size": 32,
"n_workers": 8,
"valid_steps": 2000,#2000
"warmup_steps": 1000, #1000
"save_steps": 10000, #10000
"total_steps": 70000, #70000
}
2.读取训练集地址/保存模型地址/test数据集预测结果
def parse_args():
"""arguments"""
config = {
"data_dir": "/kaggle/input/ml2022spring-hw4/Dataset",
"model_path": "/kaggle/working/model.ckpt",
"output_path": "/kaggle/working/output.csv",
}
return config
3.使用pip
首先将右侧internet开启
pip前记得加 !