iicd/Graph-DiT
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add config path

Browse Source
This commit is contained in:
xmuhanma
2024-09-22 15:47:51 +02:00
parent f75657ac3b
commit 9360839a35
3 changed files with 49 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
configs/config.yaml
View File

@@ -16,9 +16,12 @@ general:
final_model_chains_to_save: 1
enable_progress_bar: False
save_model: True
log_dir: '/nfs/data3/hanzhang/nasbenchDiT'
log_dir: '/zhome/academic/HLRS/xmu/xmuhanma/nasbenchDiT'
number_checkpoint_limit: 3
type: 'Trainer'
nas_201: '/zhome/academic/HLRS/xmu/xmuhanma/nasbenchDiT/graph_dit/NAS-Bench-201-v1_1-096897.pth'
swap_result: '/zhome/academic/HLRS/xmu/xmuhanma/nasbenchDiT/graph_dit/swap_results.csv'
root: '/zhome/academic/HLRS/xmu/xmuhanma/nasbenchDiT/graph_dit/'
model:
type: 'discrete'
transition: 'marginal'