Skip to content

Conversation

@wouterzwerink
Copy link

Fixes #316

Now, when running the example code in the issue and printing args:

{'fname': '/home/ec2-user/ffcv/tmp.beton',
 'batch_size': 100,
 'num_workers': 1,
 'os_cache': 0,
 'order': <OrderOption.RANDOM: 2>,
 'distributed': False,
 'seed': 1500678110,
 'indices': None,
 'pipelines': {'value': [<ffcv.fields.rgb_image.RandomResizedCropRGBImageDecoder at 0x7fc3bc9523a0>,
   <ffcv.transforms.flip.RandomHorizontalFlip at 0x7fc3bc952550>,
   <ffcv.transforms.ops.ToTensor at 0x7fc3bc952940>,
   <ffcv.transforms.ops.ToDevice at 0x7fc3bc9527c0>,
   <ffcv.transforms.ops.ToTorchImage at 0x7fc3bc952b50>,
   <ffcv.transforms.normalize.NormalizeImage at 0x7fc3bc952220>],
  'index': [<ffcv.fields.basics.IntDecoder at 0x7fc3bc952190>,
   <ffcv.transforms.ops.ToTensor at 0x7fc3bc975550>,
   <ffcv.transforms.common.Squeeze at 0x7fc3bc951ca0>,
   <ffcv.transforms.ops.ToDevice at 0x7fc3bc951490>]},
 'drop_last': True,
 'batches_ahead': 3,
 'recompile': False}

@andrewilyas

@andrewilyas andrewilyas merged commit 7c3e270 into libffcv:v1.1.0 Jun 29, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants