Note that better sinusoid results can be achieved by using a larger network.
"""
import csv
import numpy as np
import pickle
import random
import tensorflow.compat.v1 as tf
from data_generator import DataGenerator
from maml import MAML
from tensorflow.compat.v1.python.platform import flags
FLAGS = flags.FLAGS
## Dataset/method options
flags.DEFINE_string('datasource', 'sinusoid',
'sinusoid or omniglot or miniimagenet')
flags.DEFINE_string('expt_number', '0', '1,2,3 etc..')
flags.DEFINE_integer(
'num_classes', 5,
'number of classes used in classification (e.g. 5-way classification).')
# oracle means task id is input (only suitable for sinusoid)
flags.DEFINE_string('baseline', None, 'oracle, or None')
## Training options
flags.DEFINE_integer('pretrain_iterations', 0,
'number of pre-training iterations.')
flags.DEFINE_integer(
'metatrain_iterations', 15000,
'number of metatraining iterations.') # 15k for omniglot, 50k for sinusoid
flags.DEFINE_bool('m_metatrain', False, 'use multi-task meta training')
flags.DEFINE_integer('meta_batch_size', 25,
Note that better sinusoid results can be achieved by using a larger network.
"""
import csv
import numpy as np
import pickle
import random
import tensorflow.compat.v1 as tf
from data_generator import DataGenerator
from maml import MAML
from tensorflow.compat.v1.python.platform import flags
FLAGS = flags.FLAGS
## Dataset/method options
flags.DEFINE_string('datasource', 'sinusoid',
'sinusoid or omniglot or miniimagenet')
flags.DEFINE_string('task_setting', 'e', 'e or ne')
flags.DEFINE_integer(
'num_classes', 5,
'number of classes used in classification (e.g. 5-way classification).')
# oracle means task id is input (only suitable for sinusoid)
flags.DEFINE_string('baseline', None, 'oracle, or None')
## Training options
flags.DEFINE_integer('pretrain_iterations', 0,
'number of pre-training iterations.')
flags.DEFINE_integer(
'metatrain_iterations', 15000,
'number of metatraining iterations.') # 15k for omniglot, 50k for sinusoid
flags.DEFINE_integer('meta_batch_size', 25,
'number of tasks sampled per meta-update')