added training script note

The behavior is not obvious to me. Perhaps it's useful to mention this here to avoid confusion.
3 years ago · e710cc041f
parent 1d8ada359a
commit e710cc041f
1 changed files with 3 additions and 1 deletions
--- a/docs/scripts.md
+++ b/docs/scripts.md
@ -14,6 +14,8 @@ To train an SE-ResNet34 on ImageNet, locally distributed, 4 GPUs, one process pe

 NOTE: It is recommended to use PyTorch 1.9+ w/ PyTorch native AMP and DDP instead of APEX AMP. `--amp` defaults to native AMP as of timm ver 0.4.3.  `--apex-amp` will force use of APEX components if they are installed.

+NOTE: For training and validation the same input size is used. Some models specify a different test input size which is only used in the validation script below.
+ 
 ## Validation / Inference Scripts

 Validation and inference scripts are similar in usage. One outputs metrics on a validation set and the other outputs topk class ids in a csv. Specify the folder containing validation images, not the base as in training script.