Closed gary-kaitung closed 4 years ago
Oh, my bad, I forgot to set the bs to smaller. Problem solved.
Oh, my bad, I forgot to set the bs to smaller. Problem solved.
I know it's an old thing, but I am searching for the same thing as you - could you try to explain me what bs and sz means and what value have you set?
bs = "batch size" (the number of images trained during a single iteration of training) and sz is the length of each side of the image that you're resizing them to for training. sz=64, for example would mean you're resizing to 64x64 pixel squares for the images.
Generally I try to maximize batch size to fit the gpu's memory constraints. This version of DeOldify was developed using a single 11GB video card. So if you have something smaller than that you're going to have to adjust it accordingly.
Hi,
while running
learn_gen.fit_one_cycle(1, pct_start=0.8, max_lr=slice(1e-3))
I encountered error as below. My changes to the ipython file are My fastai version is 1.0.51
My pytorch version is 1.0.1.post2
Changed the path to
path = Path('data/manga') path_hr = path path_lr = path/'bandw'
And added about 20 pictures to this file and bandw files
I tried adding num_workers = 0 according to https://forums.fast.ai/t/brokenpipeerror-using-jupyter-notebook-lesson-1/41090/11 Although its error is about BrokenPipeError
def get_data(bs:int, sz:int, keep_pct:float): return get_colorize_data(sz=sz, bs=bs, crappy_path=path_lr, good_path=path_hr, random_seed=None, keep_pct=keep_pct, num_workers=0)
I am running the file on paperspace with very limited free GPU as trial run. Could this be memory issue? But if so, wouldn't the error tell me its about memory?
`--------------------------------------------------------------------------- StopIteration Traceback (most recent call last)