Skip to main content

Browse the archive

Revision 99f8010e483d1564144b8eedc29d8f5c432793c0 authored by Amanpreet Singh on 17 February 2021, 22:52:49 UTC, committed by Facebook GitHub Bot on 17 February 2021, 22:54:16 UTC 
[feat] Add batch_size_per_device training option (#780)
 
Summary:
Pull Request resolved: https://github.com/facebookresearch/mmf/pull/780

This will allow specifying batch size per device instead of default global batch size. If specified using `training.batch_size_per_device`, this will supersede the value specified in `training.batch_size` and information will be logged.

Reviewed By: madian9, vedanuj

Differential Revision: D26480897

fbshipit-source-id: 9ffb3d9b79a84c39db4000e6a3d051e3bd1b6eb5
1 parent daeaa95
Raw File
.pre-commit-config.yaml 
exclude: 'build'

default_language_version:
    python: python3

repos:
-   repo: https://github.com/pre-commit/pre-commit-hooks
    rev: v2.5.0
    hooks:
    -   id: trailing-whitespace
    -   id: check-ast
    -   id: check-merge-conflict
    -   id: no-commit-to-branch
        args: ['--branch=master']
    -   id: check-added-large-files
        args: ['--maxkb=500']
    -   id: end-of-file-fixer

-   repo: https://github.com/Lucas-C/pre-commit-hooks
    rev: v1.1.7
    hooks:
    -   id: insert-license
        files: \.py$
        args:
        - --license-filepath
        - docs/license_header.txt

-   repo: https://gitlab.com/pycqa/flake8
    rev: 3.7.9
    hooks:
    -   id: flake8

-   repo: https://github.com/asottile/pyupgrade
    rev: v2.2.1
    hooks:
    -   id: pyupgrade
        args: ['--py36-plus']

-   repo: https://github.com/timothycrosley/isort
    rev: 4.3.21
    hooks:
    -   id: isort
        additional_dependencies: [toml]
        exclude: ^website

-   repo: https://github.com/ambv/black
    rev: 19.3b0
    hooks:
    - id: black
      language_version: python3.7
back to top