帶網格搜索的火車

Question

我想在網格搜索期間測試火車的使用情況，但不清楚如何這樣做。

from trains import Task 
Task.init(project_name="project name", task_name='name')

在演示服務器中創建一個實驗並記錄所有內容，但無論“task_name”和

from trains import Task 
Task.create(project_name="project name", task_name='name')

可以使用不同的“task_name”調用，但不會將任何數據記錄到服務器中，而只會創建“草稿”。

這是一個示例代碼：

 epochs=[160,300]
 for epoch in epochs:
    model = define_model_run(epoch)
    model.fit(x_train,y_train)
    score = model.score(...)

我的最后一次嘗試是：

 epochs=[160,300]
 task=Task.init(project_name="demo", task_name='search')
 for epoch in epochs:
    task.create(project_name="demo", task_name=f'search_{epoch}')
    model = define_model_run(epoch)
    model.fit(x_train,y_train)
    score = model.score(...)

它記錄了實驗選項卡下的所有信息，而沒有記錄在“草稿”下。 我嘗試了最后兩個小時閱讀提供的一些文檔並閱讀源代碼，但沒有運氣。

有什么幫助嗎？

Answer 1

通過查看其 GitHub 存儲庫中已關閉的錯誤來解決： 鏈接

 epochs=[160,300]
 for epoch in epochs:
    task=Task.init(project_name="demo", task_name=f'search_{epoch}')
    model = define_model_run(epoch)
    model.fit(x_train,y_train)
    score = model.score(...)
    task.close()

那么每個任務都是“演示”項目中的不同實驗。

Answer 2

聲明：我是 TRAINS 團隊的成員

是的，這正是答案。 這個想法是你總是有一個主任務，為了創建一個新任務，你需要關閉正在運行的任務，並用一個新名稱重新初始化。 如此迅速地解決它的榮譽:)

順便說一句：您可以在此處/ 此處查看示例，展示如何發送准確度日志，以便更輕松地比較實驗，尤其是在運行超參數搜索時。