Jingyu Wang
Jingyu Wang
前端小白弱弱问一哈，这个demo是如何运行起来的呢
> 您好！ 我想问一下，每次训练得到的模型，得到的结果，差别都很大。（是不是因为 每次都是随机在一张图里cut，paste，导致的？）， > > 比如 bottle这个类，我用相同的参数分别训练两次，得到两个模型model1和model2。 测试的时候，model1 得到的结果是AUC1 = 99.444, model2得到的结果是AUC2 = 99.841. 虽然有误差，但是这相差的也太多了。。。。。 有什么办法可以改善吗？ > > 谢谢 铁汁，应该是初始化的问题。按照上面老哥的说法，设定一个固定种子一般就可以避免。 歪果仁能看到你的冲文已经很不错咯~
**well, i find the following information:** https://stackoverflow.com/questions/59249563/runtimeerror-module-must-have-its-parameters-and-buffers-on-device-cuda1-devi **just modify the code as following:** #device = torch.device("cuda:1" if torch.cuda.is_available() else "cpu") #logger.info(f"Going to train on {device}") device_ids = [1, 4, 5]...
out of memory with 40G A100 GPU
