Smadx / USTC-ML24-Fall

10 stars 2 forks source link

[lab1]关于代码框架的疑问 #2

Open ZhengWwwq opened 3 days ago

ZhengWwwq commented 3 days ago
  1. 在eval_LinearRegression函数中,注释中给出的要求返回值是两个变量,$mu_pred$ 和$relative error$,但在eval_R.py的54,55行中,只接受了返回的$mu$ image image

2.在utils.py的DataLoader类中的next方法中,在train模式下,似乎没有raise StopIteration(那行注释是我自己加的),这样就会在for batch in self.train_loader时,无限循环(可以通过加一层判断来终止) image

Smadx commented 3 days ago

1

已修复,请同步仓库

2

这里在train模式下Dataloader会无限循环是因为我们使用训练步数来控制迭代,而不是epoch-batch模式,请检查TrainerR.train()while循环条件,不必使用下面的语句:

for batch in self.train_loader:

请直接使用Dataloadernext方法