U
|
`á ã @ s d Z ddlZddlZddlZddlZddlmZ ddlmZ ddl m
|
Z
|
ddl m Z ddl mZ ddlmZ ddlZd d
|
Zd d ZdS ) ab
|
This file contains components with some default boilerplate logic user may need
|
in training / testing. They will not work for everyone, but many users may find them useful.
|
The behavior of functions/classes in this file is subject to change,
|
since they are meant to represent the "common default behavior" people need in their projects.
|
é N)Úcomm)Ú seed_all_rng)Ú PathManager)Ú setup_logger)Úcollect_env_info)Ú OrderedDictc C s t jdd} | jddddd | jdd d
|
d | jd d d d | jdd dd | jdtddd | jdtddd | jdddd | jddd d | jd ddd | jd dd d | jd td d!d d"ttjd#krÚt ¡ ndd$ }| jd%d& |¡d' | jd(d)d*t j
|
d+ | S ),zz
|
Create a parser with some common arguments used by fastreid users.
|
Returns:
|
argparse.ArgumentParser:
|
zfastreid Training)Ú descriptionz --config-fileÚ ZFILEzpath to config file)ÚdefaultÚmetavarÚhelpz
|
--finetuneÚ
|
store_truez5whether to attempt to finetune from the trained model)Úactionr z--resumez:whether to attempt to resume from the checkpoint directoryz --eval-onlyzperform evaluation onlyz
|
--num-gpusé z number of gpus *per machine*)Útyper
|
r z--num-machinesztotal number of machinesz--img_a1z1.jpgz input image)r
|
r z--img_a2z2.jpgz input image2z--img_b1z--img_b2z--machine-rankr z-the rank of this machine (unique per machine)i À Úwin32i @ z
|
--dist-urlztcp://127.0.0.1:{})r
|
Úoptsz,Modify config options using the command-lineN)r r
|
Únargs) ÚargparseÚArgumentParserÚ add_argumentÚintÚhashÚsysÚplatformÚosÚgetuidÚformatZ REMAINDER)ÚparserÚport© r ú./engine/defaults.pyÚdefault_argument_parser sD ýý ÿ"ür" c C sB | j }t ¡ r |r t |¡ t ¡ }t||dd t||d}| d |t ¡ ¡¡ | dt
|
¡ | dt | ¡ t |dr¬|j dkr¬| d |j t |j d
|
¡ ¡ ¡¡ | d | ¡¡ t ¡ r|rtj |d ¡}t |d ¡}| | ¡ ¡ W 5 Q R X | d tj |¡¡¡ t t |dr2|js>| jtjj_ dS )az
|
Perform some basic common setups at the beginning of a job, including:
|
1. Set up the detectron2 logger
|
2. Log basic information about environment, cmdline arguments, and config
|
3. Backup the config to the output directory
|
Args:
|
cfg (CfgNode): the full config to be used
|
args (argparse.NameSpace): the command line arguments to be logged
|
Zfvcore)Údistributed_rankÚname)r# z+Rank of current process: {}. World size: {}zEnvironment info:
|
zCommand line arguments: Ú config_filer z#Contents of args.config_file={}:
|
{}Úrz Running with full config:
|
{}z config.yamlÚwzFull config saved to {}Ú eval_onlyN) Ú
|
OUTPUT_DIRr Úis_main_processr ÚmkdirsÚget_rankr Úinfor Úget_world_sizer ÚstrÚhasattrr% ÚopenÚreadr ÚpathÚjoinÚwriteÚdumpÚabspathr r( ÚCUDNN_BENCHMARKÚtorchÚbackendsÚcudnnÚ benchmark)ÚcfgÚargsZ
|
output_dirÚrankÚloggerr3 Úfr r r! Ú default_setupG s2
|
|
ÿÿrB )Ú__doc__r Úloggingr r Úutilsr Z utils.envr Ú utils.file_ior Z utils.loggerr Zutils.collect_envr Ú collectionsr r9 r" rB r r r r! Ú<module> s *
|
