c1tr0n75 commited on
Commit
e8e1079
·
verified ·
1 Parent(s): ee6cc2b

Uploading model weights and training config

Browse files
training_outputs/final_model.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ef3d2d94a3e525fde1fdf61344953f9bca59d1a0a07c5b5eda9036b7a2048a97
3
+ size 109603958
training_outputs/training_config.json ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_config": {
3
+ "voxel_dim": [
4
+ 32,
5
+ 32,
6
+ 32
7
+ ],
8
+ "input_channels": 3,
9
+ "env_feature_dim": 512,
10
+ "pos_feature_dim": 64,
11
+ "hidden_dim": 256,
12
+ "num_heads": 8,
13
+ "num_layers": 4,
14
+ "max_sequence_length": 100,
15
+ "num_actions": 6
16
+ },
17
+ "training_config": {
18
+ "batch_size": 8,
19
+ "learning_rate": 0.0001,
20
+ "num_epochs": 100,
21
+ "dataset_path": "./sample_dataset",
22
+ "device": "cuda",
23
+ "total_parameters": 9113866,
24
+ "trainable_parameters": 9113866
25
+ },
26
+ "dataset_info": {
27
+ "total_samples": 1000,
28
+ "train_samples": 800,
29
+ "val_samples": 200
30
+ },
31
+ "timestamp": "2025-08-09T22:53:45.166902"
32
+ }