Skip to content

Commit

Permalink
update hellokan
Browse files Browse the repository at this point in the history
  • Loading branch information
KindXiaoming committed Aug 11, 2024
1 parent fdb60a1 commit 1c21095
Show file tree
Hide file tree
Showing 27 changed files with 303 additions and 46 deletions.
42 changes: 19 additions & 23 deletions .ipynb_checkpoints/hellokan-checkpoint.ipynb

Large diffs are not rendered by default.

42 changes: 19 additions & 23 deletions hellokan.ipynb

Large diffs are not rendered by default.

Binary file added model/0.0_cache_data
Binary file not shown.
41 changes: 41 additions & 0 deletions model/0.0_config.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,41 @@
affine_trainable: false
auto_save: true
base_fun_name: silu
ckpt_path: ./model
device: cuda
grid: 3
grid_eps: 0.02
grid_range:
- -1
- 1
k: 3
mult_arity: 2
round: 0
sb_trainable: true
sp_trainable: true
state_id: 0
symbolic.funs_name.0:
- - '0'
- '0'
- - '0'
- '0'
- - '0'
- '0'
- - '0'
- '0'
- - '0'
- '0'
symbolic.funs_name.1:
- - '0'
- '0'
- '0'
- '0'
- '0'
symbolic_enabled: true
width:
- - 2
- 0
- - 5
- 0
- - 1
- 0
Binary file added model/0.0_state
Binary file not shown.
Binary file added model/0.1_cache_data
Binary file not shown.
41 changes: 41 additions & 0 deletions model/0.1_config.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,41 @@
affine_trainable: false
auto_save: true
base_fun_name: silu
ckpt_path: ./model
device: cuda
grid: 3
grid_eps: 0.02
grid_range:
- -1
- 1
k: 3
mult_arity: 2
round: 0
sb_trainable: true
sp_trainable: true
state_id: 1
symbolic.funs_name.0:
- - '0'
- '0'
- - '0'
- '0'
- - '0'
- '0'
- - '0'
- '0'
- - '0'
- '0'
symbolic.funs_name.1:
- - '0'
- '0'
- '0'
- '0'
- '0'
symbolic_enabled: false
width:
- - 2
- 0
- - 5
- 0
- - 1
- 0
Binary file added model/0.1_state
Binary file not shown.
Binary file added model/0.2_cache_data
Binary file not shown.
29 changes: 29 additions & 0 deletions model/0.2_config.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,29 @@
affine_trainable: false
auto_save: true
base_fun_name: silu
ckpt_path: ./model
device: cuda
grid: 3
grid_eps: 0.02
grid_range:
- -1
- 1
k: 3
mult_arity: 2
round: 0
sb_trainable: true
sp_trainable: true
state_id: 2
symbolic.funs_name.0:
- - '0'
- '0'
symbolic.funs_name.1:
- - '0'
symbolic_enabled: true
width:
- - 2
- 0
- - 1
- 0
- - 1
- 0
Binary file added model/0.2_state
Binary file not shown.
Binary file added model/0.3_cache_data
Binary file not shown.
29 changes: 29 additions & 0 deletions model/0.3_config.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,29 @@
affine_trainable: false
auto_save: true
base_fun_name: silu
ckpt_path: ./model
device: cuda
grid: 3
grid_eps: 0.02
grid_range:
- -1
- 1
k: 3
mult_arity: 2
round: 0
sb_trainable: true
sp_trainable: true
state_id: 3
symbolic.funs_name.0:
- - '0'
- '0'
symbolic.funs_name.1:
- - '0'
symbolic_enabled: false
width:
- - 2
- 0
- - 1
- 0
- - 1
- 0
Binary file added model/0.3_state
Binary file not shown.
Binary file added model/0.4_cache_data
Binary file not shown.
29 changes: 29 additions & 0 deletions model/0.4_config.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,29 @@
affine_trainable: false
auto_save: true
base_fun_name: silu
ckpt_path: ./model
device: cuda
grid: 3
grid_eps: 0.02
grid_range:
- -1
- 1
k: 3
mult_arity: 2
round: 0
sb_trainable: true
sp_trainable: true
state_id: 4
symbolic.funs_name.0:
- - '0'
- '0'
symbolic.funs_name.1:
- - '0'
symbolic_enabled: true
width:
- - 2
- 0
- - 1
- 0
- - 1
- 0
Binary file added model/0.4_state
Binary file not shown.
Binary file added model/0.5_cache_data
Binary file not shown.
29 changes: 29 additions & 0 deletions model/0.5_config.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,29 @@
affine_trainable: false
auto_save: true
base_fun_name: silu
ckpt_path: ./model
device: cuda
grid: 10
grid_eps: 0.02
grid_range:
- -1
- 1
k: 3
mult_arity: 2
round: 0
sb_trainable: true
sp_trainable: true
state_id: 5
symbolic.funs_name.0:
- - '0'
- '0'
symbolic.funs_name.1:
- - '0'
symbolic_enabled: false
width:
- - 2
- 0
- - 1
- 0
- - 1
- 0
Binary file added model/0.5_state
Binary file not shown.
Binary file added model/0.6_cache_data
Binary file not shown.
29 changes: 29 additions & 0 deletions model/0.6_config.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,29 @@
affine_trainable: false
auto_save: true
base_fun_name: silu
ckpt_path: ./model
device: cuda
grid: 10
grid_eps: 0.02
grid_range:
- -1
- 1
k: 3
mult_arity: 2
round: 0
sb_trainable: true
sp_trainable: true
state_id: 6
symbolic.funs_name.0:
- - sin
- x^2
symbolic.funs_name.1:
- - exp
symbolic_enabled: true
width:
- - 2
- 0
- - 1
- 0
- - 1
- 0
Binary file added model/0.6_state
Binary file not shown.
Binary file added model/0.7_cache_data
Binary file not shown.
29 changes: 29 additions & 0 deletions model/0.7_config.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,29 @@
affine_trainable: false
auto_save: true
base_fun_name: silu
ckpt_path: ./model
device: cuda
grid: 10
grid_eps: 0.02
grid_range:
- -1
- 1
k: 3
mult_arity: 2
round: 0
sb_trainable: true
sp_trainable: true
state_id: 7
symbolic.funs_name.0:
- - sin
- x^2
symbolic.funs_name.1:
- - exp
symbolic_enabled: true
width:
- - 2
- 0
- - 1
- 0
- - 1
- 0
Binary file added model/0.7_state
Binary file not shown.
9 changes: 9 additions & 0 deletions model/history.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,9 @@
### Round 0 ###
init => 0.0
0.0 => fit => 0.1
0.1 => prune => 0.2
0.2 => fit => 0.3
0.3 => refine => 0.4
0.4 => fit => 0.5
0.5 => auto_symbolic => 0.6
0.6 => fit => 0.7

0 comments on commit 1c21095

Please sign in to comment.