expert_trajectory
2023-08-11 16:34:33 0 举报Code Understanding of: Dataset Distillation by Matching Training Trajectories - buffer.py
作者其他创作
大纲/内容
dst_train
初始化teacher_net
更新
teacher_net
args.dc_aug_param
epoch()
trajectory
teacher_net 训练循环num_experts 次
input args
get_dataset()
all args
创建交叉熵 loss function
整理后的dst_train
设置 trajectory的save_dir
当前 epoch在 lr 更新 schedule中
utils
testloader
criterion
get_daparam()
teacher_optim
get_network()
load dataset (ZCA-ed)
img 新增一维，label 映射为 idx
save_dir
lr_schedule
save trajectory
timestamps
images_alllabels_all
buffer
train_epoch 次
trainloader
学习率衰减
收藏
0 条评论
下一页
