Skip to content

pytorch mini batch size

2019.12.19 06:12

WHRIA 조회 수:82

https://stackoverflow.com/questions/52518324/how-to-compensate-if-i-cant-do-a-large-batch-size-in-neural-network/52523847

 

 

4

In pytorch, when you perform the backward step (calling loss.backward() or similar) the gradients are accumulated in-place. This means that if you call loss.backward() multiple times, the previously calculated gradients are not replaced, but in stead the new gradients get added on to the previous ones. That is why, when using pytorch, it is usually necessary to explicitly zero the gradients between minibatches (by calling optimiser.zero_grad() or similar).

If your batch size is limited, you can simulate a larger batch size by breaking a large batch up into smaller pieces, and only calling optimiser.step() to update the model parameters after all the pieces have been processed.

For example, suppose you are only able to do batches of size 64, but you wish to simulate a batch size of 128. If the original training loop looks like:

optimiser.zero_grad()
loss = model(batch_data) # batch_data is a batch of size 128
loss.backward()
optimiser.step()

then you could change this to:

optimiser.zero_grad()

smaller_batches = batch_data[:64], batch_data[64:128]
for batch in smaller_batches:
    loss = model(batch) / 2
    loss.backward()

optimiser.step()

and the updates to the model parameters would be the same in each case (apart maybe from some small numerical error). Note that you have to rescale the loss to make the update the same.

번호 제목 글쓴이 날짜 조회 수
1716 mAP 계산법 WHRIA 2019.02.24 45558
1715 epitope spreading han 2006.05.09 40968
1714 encfs WHRIA 2020.05.17 33689
1713 Heinrich Law (1:29:300 Law) WHRIA 2007.08.12 24963
1712 일본 주소 [2] WHRIA 2008.06.28 24013
1711 사주팔자 프로그램 file 한승석 2003.02.16 20304
1710 simple adblock file WHRIA 2012.04.13 18071
1709 도란사민 WHRIA 2011.04.19 18067
1708 탤런트 이윤지씨와 함께 file WHRIA 2010.02.04 17006
1707 penicillin han 2003.12.10 15672
1706 SSH tunnel WHRIA 2007.10.01 15078
1705 geexbox [1] han 2006.12.01 14601
1704 아이피부과 개원 WHRIA 2010.01.18 14407
1703 세무회계 WHRIA 2010.01.31 14380
1702 증명사진 WHRIA 2010.03.14 14308

Powered by Xpress Engine / Designed by Sketchbook

sketchbook5, 스케치북5

sketchbook5, 스케치북5

나눔글꼴 설치 안내


이 PC에는 나눔글꼴이 설치되어 있지 않습니다.

이 사이트를 나눔글꼴로 보기 위해서는
나눔글꼴을 설치해야 합니다.

설치 취소