mirrors
/
deit
mirror of https://github.com/facebookresearch/deit
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

update wandb cfg

pull/222/head
fabfish 2023-05-29 21:18:21 +08:00
parent bbf3c5fae1
commit 239eb9bc78
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
main.py
View File

@ -30,6 +30,11 @@ import utils
import wandb
import os
os.environ['WANDB_DIR'] = os.getcwd() + "/wandb/"
os.environ['WANDB_CACHE_DIR'] = os.getcwd() + "/wandb/.cache/"
os.environ['WANDB_CONFIG_DIR'] = os.getcwd() + "/wandb/.config/"
def get_args_parser():
parser = argparse.ArgumentParser('DeiT training and evaluation script', add_help=False)