❓问题
如何实现本地存储工件，例如模型检查点？
就我所知( https://aimstack.readthedocs.io/en/latest/using/artifacts.html ),我们可以使用 run.log_artifact() 将已经在磁盘上的工件上传到某个地方。这对于S3/远程存储是有意义的，但对于本地存储呢？有没有一种方法可以直接将工件存储在磁盘上？基本上，我想每隔n个epoch保存一次模型的检查点。
类似于PyTorch中的代码：
log_path = self.run.get_this_from_somewhere() # Get path for current run
torch.save(self.state_dict(), os.path.join(log_path, f"model_{epoch}.pt"))
self.run.log_artifact(os.path.join(log_path, f"model_{epoch}.pt"), name=f"model_{epoch}.pt") # without upload
2条答案
按热度按时间8cdiaqws1#
将工件存储在本地的能力将会非常有用。
q7solyqu2#
尝试了一下这个 - #3156 。希望得到反馈。