Skip to content

Commit 20b2172

Browse files
CydraldaviskingCopilotAldric PIERRAIN
authored
Implementation of ARC-AGI dataset loader (#3115)
* Implementation of linear_ layer for neural networks. This layer provides an optimized linear transformation for multi-dimensional inputs. * Minor change * Update dlib/dnn/layers.h Co-authored-by: Copilot <[email protected]> * Add reshape_to and flatten layers to Dlib's DNN module * Missing update to "visitors.h" * format fixing for reshape_to * Update dlib/test/dnn.cpp * Vocabulary size fixed for learning, and function added for transformation-free tokenization * Added a new example for learning a “complex” Transformer model. * Added a new example for learning a “complex” Transformer model. * Updated example for training a Transformer model. * fix for gcc/ffmpeg compilation * Fix a warning message for Ubuntu compilation. * Update for Linux environment. * Fix batch building * Slight improvement in model definition. * linear_ layer implementation improvement * finalizing the example * Fixing break condition in training method. * Fixing declaration order of variables. * bpe_tokenizer improvements. * Example updated. * bpe_tokenizer class refactoring. * Example updated. * bpe_tokenizer class updated. * Decoding part of the bpe_tokenizer updated. * Network definition update * Add Adaptive Computation Time (ACT) layer with CPU/CUDA support * Fixes * Update comments for params * Fixes and improvements * Disabling enable_depth_scaling, which obviously affects the result of test_layer * Implementation of ARC-AGI dataset loader --------- Co-authored-by: Davis E. King <[email protected]> Co-authored-by: Copilot <[email protected]> Co-authored-by: Aldric PIERRAIN <[email protected]>
1 parent 2ac06f9 commit 20b2172

File tree

3 files changed

+1212
-0
lines changed

3 files changed

+1212
-0
lines changed

dlib/data_io.h

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
#include "data_io/image_dataset_metadata.h"
88
#include "data_io/mnist.h"
99
#include "data_io/cifar.h"
10+
#include "data_io/arc_agi.h"
1011

1112
#ifndef DLIB_ISO_CPP_ONLY
1213
#include "data_io/load_image_dataset.h"

0 commit comments

Comments
 (0)