tianlianghai

Didn't set model to eval mode when validating in chapter 8

1

In the book and code when we first met resnet, the model has some batch norm layers. But in validate function, it didn't call model.eval(), I don't think this is...

如何使用这个模型推理

5

已经在Bike数据集训练和微调好了，使用自己的数据来进行预测呢

the device used in training

1

what device did you use in training, I use 512 per V100 16GB lead to an OOM error. but if I use a small batch, the loss go to NaN...